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.]]]