您在此处指定模型输入。您希望将批处理大小保留为
None,这意味着您可以使用可变数量的输入(一个或多个)来运行模型。批处理对于有效使用您的计算资源很重要。
x = tf.placeholder("float", shape=[None, 784])y_ = tf.placeholder("float", shape=[None, 10])下一条重要的线是:
batch = mnist.train.next_batch(50)
在这里,您将发送50个元素作为输入,但您也可以将其更改为一个
batch = mnist.train.next_batch(1)
无需修改图形。如果指定批处理大小(在第一个代码段中用一些数字代替“无”),则每次都必须更改,这并不理想,特别是在生产中。



