2017-06-20 79 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」,因爲你不想完全指定一次會輸入多少圖像。