Keras - 如何通过摆脱最后三个维度将5d Tensor变成3d？

Question

在Keras，我有一个（100,200,300,400,500）Tensor从一个层输出，并希望将它变成一个（100,200,300 * 400 * 500），然后将其输入新图层。我该怎么做？

Answer 1

尝试添加keras.layers.Reshape图层：

from keras.layers import Reshape

# ...
model.add(Reshape((100,200,300*400*500)))
# ...

或

model.add(Reshape((-1,200,300*400*500)))

Answer 2

假设第一个维度是批量大小，则忽略它：

model.add(Reshape(200,300*400*500))