mindspore.ops.Transpose

class mindspore.ops.Transpose[源代码]

根据指定的排列对输入的Tensor进行数据重排。

更多参考详见 mindspore.ops.transpose()

支持平台:

Ascend GPU CPU

样例:

>>> input_x = Tensor(np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]]), mindspore.float32)
>>> input_perm = (0, 2, 1)
>>> transpose = ops.Transpose()
>>> output = transpose(input_x, input_perm)
>>> print(output)
[[[ 1.  4.]
  [ 2.  5.]
  [ 3.  6.]]
 [[ 7. 10.]
  [ 8. 11.]
  [ 9. 12.]]]