tf.placeholder_with_default导致错误

Question

简短的问题。 为什么导致此行错误？ ：

x = tf.placeholder_with_default([0.0 for _ in range(784)], [None, 784], name='images')

我该如何解决？ 提前谢谢。

发生此错误：

ValueError: Shapes must be equal rank, but are 1 and 2 for 'images' (op: 'PlaceholderWithDefault') with input shapes: [784].

Answer 1

问题是您为占位符创建了[None，784]形状，但默认值具有形状[784]。因此，只需在默认值周围添加一个方括号，例如：

x = tf.placeholder_with_default([[0.0 for _ in range(784)]], [None, 784], name='images')

所以你的形状为[1,784]