Tensorflow conv2d_transpose 大小错误“out_backprop 的行数与计算的不匹配”

2024-05-16

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

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'))

where

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

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

我缺少什么?


发现错误了。填充应该是“有效”,而不是“相同”。

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

Tensorflow conv2d_transpose 大小错误“out_backprop 的行数与计算的不匹配” 的相关文章

随机推荐