KERAS-如何通过摆脱最后三个维度将5D张量变成3D

Question

在keras中，我有一个(100、200、300、400、500(从一层输出的张量，并希望将其变成(100、200、300*400*(500(将其喂入新层之前。我该怎么做？

Answer 1

尝试添加keras.layers.reshape layer：

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))