python - 在 Keras 中，如何使用无维度的 Reshape 层？

Question

在我的模型中，层的形状为 [None, None, 40, 64]。我想将其 reshape 为 [None, None, 40*64]。但是，如果我只是执行以下操作:

reshaped_layer = Reshape((None, None, 40*64))(my_layer)

它抛出一个错误，提示 None values not supported。

(需要说明的是，这不是 tf.keras，这只是 Keras)。

Answer 1

首先，您传递给 Reshape 层的参数是批处理中一个样本的所需形状，而不是整批样本。因此，由于批处理中的每个样本都是 3D 张量，因此参数还必须仅考虑该 3D 张量(即不包括批处理轴)。

其次，您可以使用-1 作为只有一个轴的形状。它告诉 Reshape 层根据您提供的其他轴的形状自动推断该轴的形状。所以考虑到这两点，那就是:

reshaped_out = Reshape((-1, 40*64))(layer_out)