Tensorflow conv2d_transpose size error“out_backprop的行数与计算不匹配”

时间:2016-05-01 14:41:23

标签: machine-learning tensorflow

我正在tensorflow中创建一个卷积自动编码器。我收到了这个确切的错误:

tensorflow.python.framework.errors.InvalidArgumentError: Conv2DBackpropInput: Number of rows of out_backprop doesn't match computed: actual = 8, computed = 12
     [[Node: conv2d_transpose = Conv2DBackpropInput[T=DT_FLOAT, data_format="NHWC", padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/cpu:0"](conv2d_transpose/output_shape, Variable_1/read, MaxPool_1)]]

相关代码:

l1d = tf.nn.relu(tf.nn.conv2d_transpose(l1da, w2, [10, 12, 12, 32], strides=[1, 1, 1, 1], padding='SAME'))

其中

w2 = tf.Variable(tf.random_normal([5, 5, 32, 64], stddev=0.01))

我检查了输入到conv2d_transpose的形状,即l1da,它是正确的(10x8x8x64)。批量大小为10,此层的输入为8x8x64,输出为12x12x32。

我错过了什么?

1 个答案:

答案 0 :(得分:4)

发现错误。填充应该是"有效",不是"相同"。