我正在尝试将一个图层的输出传递并连接到CNN中的另一个图层,如下图所示。重塑后,30x30x512层与15x15x1024连接。
请注意,两种不同的型号并合并它们并不是我想要的解决方案。我想知道如何使用相同的模型来合并提到的图层。
提前感谢您的建议。
答案 0 :(得分:3)
您需要存储要合并的两个图层的输出张量。拥有这些张量后,至少有两种方法可以合并它们:
使用Reshape
图层,您可以在其中指定输出形状并完成作业。
使用tf.space_to_depth
函数并将其打包到Lambda
图层(因为它是TensorFlow操作)。这可能仅在您使用TensorFlow后端时使用。
重新塑造输出后,您可以使用concatenate
图层并按渠道尺寸合并输出。