Керас изменяется от (Нет, Нет, Нет, 512) до (Нет, Нет, Нет, Нет, 512)?

Question

Я пытаюсь преобразовать 4D-тензор в 5D-тензор, но Reshape не принимает None Values ​​(я пытался заменить его на -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())