整形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]) 

问候