Reshape INPUT_DATA (Form = [Keine, 28, 28, 1]) Input_Data (Form = [None, 28, 28])

Question

Ich habe eine Variable:

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

Ich will es die konvertieren über Form. Ich mache es mit Hilfe von tflearn.reshape, aber es funktioniert nicht richtig. Hier ist der Code "

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

Ich will nicht deklarieren, sie zu gestalten [None, 28,28] wie ich Elemente. Jede Hilfe, wie es zu tun?

Answer 1

Verwenden tf.squeeze es entfernt Dimensionen der Größe 1 aus der Form eines Tensor.

network = tf.squeeze(network) 

Answer 2

das Problem in der Form ist, dass Sie setzen. es sein sollte [-1,28,28] Format.

So Ihr Code sei:

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

Grüße