example - Swap tensor de ejes en tensorflow

tensorflow reshape example

Question

Tengo un tensor de forma (30,116,10) y quiero intercambiar las dos primeras dimensiones, de modo que tengo un tensor de forma (116,30,10)

Vi ese número implementado como una función (np.swapaxes) y busqué algo similar en tensorflow pero no encontré nada.

Tiene alguna idea ?

Gracias por adelantado,

Alexis

Answer 1

tf.transpose proporciona la misma funcionalidad que np.swapaxes , aunque en una forma más general. En su caso, puede hacer tf.transpose(orig_tensor, [1, 0, 2]) que sería equivalente a np.swapaxes(orig_np_array, 0, 1) .