keras重塑图像的范围从（无，无，1）到（256、256、1）

Question

我的网络采用随机大小的图像作为输入，因此输入层的形状为（None，None，1）。可以将形状转换为（256，256，1）。

我尝试过的是这个，但是没有用。

   

g_img = Reshape(target_shape=(256, 256, 1), input_shape=(None, None, 1), name='gD_input')

out = Conv2D(64, kernel_size=5, strides=2, padding='same', name='gD_conv0')(g_img)
out = BatchNormalization(name='gD_bn0')(out)
out = LeakyReLU(alpha=0.2, name='gD_ac0')(out)

有什么建议吗？