如何创建多维张量?

时间:2017-06-20 08:43:56

标签: python tensorflow deep-learning

我指的是TensorFlow网站上的MNIST教程。存在用于保持28乘28像素MNIST图像的张量。形状如下:

x = tf.placeholder(tf.float32, shape=[None, 784])

正如您所看到的,他们已经将矩阵展平并将所有像素用作784值的一维向量。

如何使用28 x 28像素的适​​当矩阵设置张量?我想保留有关图像2D结构的信息。

1 个答案:

答案 0 :(得分:1)

创建占位符张量的方法是相同的:

x = tf.placeholder(tf.float32, shape=[None, 28, 28])

但当然现在必须使用适当大小的图像而不是前面提到的扁平图像(矢量)。

编辑:注意当然你还有"无"在第一个维度,因为您不想完全指定一次输入多少图像。