2017-12-03 18 views
0

Iは変数を持っている:リシェイプINPUT_DATA(形状=なし、28、28、1])INPUT_DATAする(形状=なし、28、28])

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

私はそれを変換します上記の形状。私はtflearn.reshapeの助けを借りてそれをやっているが、それは正しく動作していない。ここで私は、私はそれの要素を持っているように、[なし]、28,28]を形作るないし、それを再宣言したくないコード」

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

がある。それを行う方法上の任意のヘルプ?

答えて

0

使用tf.squeezeそれはテンソルの形状から、大きさ1の寸法を削除します。

network = tf.squeeze(network) 
0

問題は、あなたが入れた形状である。それは[-1,28,28]の形式でなければなりません。

だからあなたのコードがしますbe:

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

よろしくお願いいたします。

関連する問題