2016-07-05 2 views
9

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)Swap tensore assi in tensorflow

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

risposta

21

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).

Problemi correlati