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