我有一个输入图层，我尝试进行零填充，使其成为TensorShape([1, 1, 104, 24])的特定维度
import tensorflow as tf
import dumpy as np
input_shape = (1, 1, 1, 24)
x = np.arange(np.prod(input_shape)).reshape(input_shape) # (1, 1, 1, 24)
y = tf.keras.layers.ZeroPadding2D(padding=(0, 51))(x)
y.shape # TensorShape([1, 1, 103, 24])
# how do I make the y.shape --> [1, 1, 104, 24]??我怎样改变y的参数，使我可以得到**[1，1，104，24]**的形状？
1条答案
按热度按时间xzabzqsa1#
您正在使用：
这意味着：（symmetric_height_pad，symmetric_width_pad）。也就是说，您要向数字的左侧和右侧添加51个零，因此您将得到51 + 1 + 51 = 103项。要得到104项，您可以使用以下命令向左侧添加51，向右添加52：
这里的数字表示：* （（顶部填充，底部填充），（左侧填充，右侧填充））*