2017-06-20 51 views
0

我指的是TensorFlow网站上的MNIST教程。有一个张量用于保存28×28像素的MNIST图像。外形看起来如下:如何创建一个多维张量?

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

正如你可以看到它们具有扁平的基体,并使用所有的像素作为784个的值的一维向量。

如何建立一个28×28像素的适​​当矩阵张量?我想保留有关图像二维结构的信息。

回答

1

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

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

但当然它现在必须与合适尺寸的图像,而不是前述的扁平图像(载体)被馈送。

编辑:当然你还是在第一维中没有“None”,因为你不想完全指定一次会输入多少图像。

相关问题