将input_data(shape = [None,28,28,1])重塑为input_data(shape = [None,28,28])

时间:2017-12-03 08:24:50

标签: python tensorflow tflearn

我有一个变量:

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

我想将其转换为上述形状。我是在tflearn.reshape的帮助下做的,但它没有正常工作。这是代码"

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

我不想重新声明它以塑造[无,28,28]因为我有元素。有关如何做的任何帮助吗?

1 个答案:

答案 0 :(得分:0)

问题在于你所处的形状。它应该是[-1,28,28]格式。

所以你的代码将是:

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

此致