【发布时间】:2016-07-08 20:12:36
【问题描述】:
考虑神经网络中每一层的批次维度似乎有点麻烦。为什么我们在 Tensorflow 中没有一些功能可以为整个模型设置批量大小?
【问题讨论】:
标签: tensorflow tensorflow-serving
考虑神经网络中每一层的批次维度似乎有点麻烦。为什么我们在 Tensorflow 中没有一些功能可以为整个模型设置批量大小?
【问题讨论】:
标签: tensorflow tensorflow-serving
在 tensorflow 中,您不必考虑批量大小。
在MNIST Tutorial 中解释了 tensorflow 如何处理各种大小的批次。
引用教程:
x = tf.placeholder(tf.float32, shape=[None, 784])
y_ = tf.placeholder(tf.float32, shape=[None, 10])
输入图像 x 将由浮点数的二维张量组成。这里我们给它分配了[None, 784]的形状,其中784是单个扁平化MNIST图像的维度,None表示第一个维度,对应于batch size,可以是任意大小。 em>
【讨论】: