假设我有两个要连接的层:(稠密的)称为outg,(Conv2D)称为outm,输出形状为(None,256)和(None,32,32,256)。我的目标是将输出合并到形状中(无、32、32、512)。为了更好地理解,请参见下面的图片。蓝色区域表示Conv2D,橙色区域表示重复的密集层输出。
我的想法是使用来自keras.layers
的RepeatVector, Reshape and Concatenate
,所以我做了:
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输出垂直,所以我的问题是:它做得对吗,还是我忽略了什么
目前没有回答
相关问题 更多 >
编程相关推荐