Swap tensore assi in tensorflow

Question

Ho un tensore di forma (30,116,10) e voglio scambiare le prime due dimesions, in modo da avere un tensore di forma (116,30,10)

ho visto che numpy come tale funzione implementata (np.swapaxes) e ho cercato qualcosa di simile in tensorflow ma non ho trovato nulla.

Avete qualche idea?

Grazie in anticipo,

Alexis

Answer 1

tf.transpose fornisce la stessa funzionalità np.swapaxes, anche se in una forma più generalizzata. Nel tuo caso, puoi fare tf.transpose(orig_tensor, [1, 0, 2]) che sarebbe equivalente a np.swapaxes(orig_np_array, 0, 1).