我指的是TensorFlow网站上的MNIST教程。存在用于保持28乘28像素MNIST图像的张量。形状如下:
x = tf.placeholder(tf.float32, shape=[None, 784])
正如您所看到的,他们已经将矩阵展平并将所有像素用作784值的一维向量。
如何使用28 x 28像素的适当矩阵设置张量?我想保留有关图像2D结构的信息。
答案 0 :(得分:1)
创建占位符张量的方法是相同的:
x = tf.placeholder(tf.float32, shape=[None, 28, 28])
但当然现在必须使用适当大小的图像而不是前面提到的扁平图像(矢量)。
编辑:注意当然你还有"无"在第一个维度,因为您不想完全指定一次输入多少图像。