python - 将 Keras Dense 连接到 Conv2D 输出

Question

假设我要连接两层：(Dense) 称为 outg 和 (Conv2D) 称为 outm，输出形状为 (None, 256) 和 (None, 32, 32, 256)。我的目标是将输出合并为形状（无、32、32、512）。下图便于理解。蓝色区域代表 Conv2D，橙色区域代表重复的 Dense 层输出。

我的想法是使用RepeatVector, Reshape and Concatenatefrom keras.layers，所以我做了：

outg_repeated = RepeatVector(32*32)(outg)
outg_reshaped = Reshape((32, 32, 256))(outg_repeated)
concatenated = Concatenate(axis=-1)([outm, outg_reshaped])

它给出了所需的形状（None, 32, 32, 512），但是我不知道我是否正确连接了它，这与 Conv2D 输出垂直，所以我的问题是：它做得对还是我忽略了什么？

标签： pythontensorflowkerasconcatenation