web-dev-qa-db-fra.com

Comment permuter les axes du tenseur dans TensorFlow?

J'ai un tenseur de forme (30, 116, 10), et je veux échanger les deux premières dimensions, pour avoir un tenseur de forme (116, 30, 10)

J'ai vu ce numpy comme une telle fonction implémentée (np.swapaxes) et j'ai cherché quelque chose de similaire dans tensorflow mais je n'ai rien trouvé.

Avez-vous une idée?

19
Alexis Rosuel

tf.transpose fournit les mêmes fonctionnalités que np.swapaxes, Bien que sous une forme plus généralisée. Dans votre cas, vous pouvez faire tf.transpose(orig_tensor, [1, 0, 2]) qui serait équivalent à np.swapaxes(orig_np_array, 0, 1).

34
keveman