整形input_data(形状= [无,28个,28,1])至input_data(形状= [无,28个,28])

我有一个变量:整形input_data(形状= [无,28个,28,1])至input_data(形状= [无,28个,28])

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

我想将其转换为以上形状。我正在做tflearn.reshape的帮助,但它不能正常工作。下面是代码“

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

我不想重新声明它塑造[无,28,28]因为我在它的元素。如何做到这一点任何帮助吗?

回答:

使用tf.squeeze它从一个张量的形状消除尺寸1的尺寸。

network = tf.squeeze(network) 

回答:

问题是出在你把形状。它应该是[-1,28,28]格式。

所以您的代码将成为:

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

问候

以上是 整形input_data(形状= [无,28个,28,1])至input_data(形状= [无,28个,28]) 的全部内容, 来源链接: utcz.com/qa/260034.html

回到顶部