Keras是否从( None，512)重塑为(None，None,512)？

Question

我正在尝试将4D张量重塑为5D张量，但reshape不接受None值(我尝试将其替换为-1，但也不起作用)您知道如何实现这一点吗？有没有可能做到这一点？

我试过了：

x=Conv2D(512, (3,3), activation='relu', padding='same')(x)

new_dims=(1,-1,-1,512)

res=Reshape(target_shape=new_dims)(x)

Answer 1

您必须根据所需的大小调整Reshape中的值。

from keras.layers import Input, Conv2D, Reshape
from keras.models import Model

input = Input((224,224,3))
x = Conv2D(512, (3,3), activation='relu', padding='same')(input)
output = Reshape((2, 112, 224, 512))(x)
model = Model(input, output)
print(model.summary())

​

​