tensorflow.python.framework.errors_impl.InvalidArgumentError:不兼容的形状:[1568]与[32,49]

时间:2018-10-28 10:47:00

标签: tensorflow keras lstm

我的keras编码器-解码器代码在cpu上工作正常。但是在gpu上,我无法使用1以外的batch_size启动它!这很奇怪。即使我在这种拟合中不使用批处理大小参数,我也会得到:

tensorflow.python.framework.errors_impl.InvalidArgumentError:不兼容的形状:[1568]与[32,49]

(默认情况下,keras的代码为batch_size = 32)

我可以使用任何我想要的batch_size运行许多其他程序。哪个模块将一个二维对象与一维对象进行比较?

1 个答案:

答案 0 :(得分:-1)

我的问题很相似,但是我得到了:

tensorflow.python.framework.errors_impl.InvalidArgumentError:
不兼容的形状:[64]与[128]。

我发现这可能是因为损失函数未执行降维操作。
因此,我尝试了tf.reduce_mean(),它显示正常。