Wie können die 3D-Tensor-Dimensionen im Tensorflow gemischt werden?

Question

Ich bin vertraut mit Theano dimshuffle Funktion, die Permutation von Abmessungen eines Tensors ermöglicht. Ich suche nach einer äquivalenten Funktion im Tensorflow (derzeit mit 0.8.0).

Bei einem dreidimensionalen Tensor x, die als x[a,b,c] indiziert ist, würde Ich mag so etwas wie x.dimshuffle(1,0,2) so tun, dass ich kann Index als [b,a,c].

Wird es unterstützt? Wenn nicht, wie kann ich dasselbe Ergebnis erzielen, indem ich bereits implementierte Methoden verwende?

Answer 1

In der Tat denke ich, dass tf.transpose() genau das tut, was Sie brauchen. Übergeben Sie einfach die gewünschte Permutation als Liste im zweiten Argument der Funktion. Für Details siehe: https://www.tensorflow.org/versions/r0.8/api_docs/python/array_ops.html#transpose