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