tensorflow - reshape 层的输出维度

Question

model = tf.keras.Sequential()
model.add(layers.Dense(7*7*256, use_bias=False, input_shape=(100,)))
model.add(layers.BatchNormalization())
model.add(layers.LeakyReLU())

model.add(layers.Reshape((7, 7, 256)))

密集层接受 1*100 维度的输入。它在其层中使用 7*7*256 个节点。 reshape 层以 1*(7*7*256) 作为输入，其输出是什么。我的意思是 (7, 7, 256) 是什么意思？

如果我们输入1*100的图像，它是7*7尺寸的图像吗？它是什么？

对不起，我知道我的理解完全错误。所以我想了解它。

Answer 1

这里你的模型将采用 (*, 100) 的 input_shape，第一个密集层将输出 (*, 7*7*256) 的形状，最后一个 Reshape 层将将该输出 reshape 为形状数组(*、7、7、256)。

其中 * 是您的批量大小。

所以基本上，你的形状 (,100) 的“图像”将被 reshape 为形状数组(, 7, 7, 256)。

希望这对您有帮助