整形input_data（形狀= [無，28個，28，1]）至input_data（形狀= [無，28個，28]）

Question

我有一個變量：

network = input_data(shape=[None, 28, 28, 1]) 

我想將其轉換爲以上形狀。我正在做tflearn.reshape的幫助，但它不能正常工作。下面是代碼「

network = tflearn.reshape(network,(-1, 28, 28)) 

我不想重新聲明它塑造[無，28,28]因爲我在它的元素。如何做到這一點任何幫助嗎？

Answer 1

使用tf.squeeze它從一個張量的形狀消除尺寸1的尺寸。

network = tf.squeeze(network) 

Answer 2

問題是出在你把形狀。它應該是[-1,28,28]格式。

所以您的代碼將成爲：

network = tflearn.reshape(network,[-1, 28, 28]) 

問候