it-swarm.com.de

Wie tausche ich Tensorachsen in TensorFlow?

Ich habe einen Formtensor (30, 116, 10) Und möchte die ersten beiden Dimensionen tauschen, sodass ich einen Formtensor (116, 30, 10) Habe.

Ich habe diese Funktion als solche implementiert gesehen (np.swapaxes) Und im Tensorflow nach etwas Ähnlichem gesucht, aber nichts gefunden.

Hast du irgendeine Idee?

19
Alexis Rosuel

tf.transpose bietet die gleiche Funktionalität wie np.swapaxes, Allerdings in einer allgemeineren Form. In Ihrem Fall können Sie tf.transpose(orig_tensor, [1, 0, 2]) ausführen, was np.swapaxes(orig_np_array, 0, 1) entspricht.

34
keveman