2017-07-27 7 views
0

I eine for-Schleife haben, zu erzeugen, die Vektoren (tf Tensoren) gleicher Länge erzeugt, sagenTensorflow Stapel Vektoren aus Schleife eine Matrix

a1 = [0, 2, 4 ... ] 
a2 = [1, 4, 6 ... ] 
... 

und ich möchte diese Vektoren in einer Matrix verketten entlang der 0-te Achse

matrix = [[0,2,4...] , [1,4,6...] ... ] 

ich kann einen tun

matrix = tf.concat(0, [matrix, a]) 

innerhalb der for-Schleife. Die erste Iteration funktioniert jedoch nicht, da die Matrix nicht existiert und wenn ich sie zu einem Vektor initialisiere, stecke ich mit diesem Vektor an der Spitze der Endmatrix fest. Gibt es einen schnellen Weg, dies zu tun?

Antwort

1

können Sie tf.stack verwenden:

matrix = tf.stack([a1, a2, ...]) 
+0

Sie meinen, hängen Sie die a1, a2 auf eine leere Liste und rufen Stapel auf der Liste? – Qubix

+0

Ja, das könnte ein Weg sein, es zu tun. – user1735003

Verwandte Themen