mindspore.ops.permute

查看源文件
mindspore.ops.permute(input, axis)[源代码]

按照输入 axis 的维度顺序排列输入Tensor。

参数:
  • input (Tensor) - 输入Tensor。

  • axis (tuple(int)) - 维度的顺序,permute根据 axis 的顺序重新排列 input

返回:

Tensor,具有和输入Tensor相同的维数,按照 axis 重新排列。

异常:
  • ValueError - axis 为None。

  • ValueError - axis 的元素总量不等于 input 的维数。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]]), mindspore.float32)
>>> input_perm = (0, 2, 1)
>>> print(ops.permute(input_x, input_perm))
[[[ 1.  4.]
  [ 2.  5.]
  [ 3.  6.]]
 [[ 7. 10.]
  [ 8. 11.]
  [ 9. 12.]]]