mindspore.ops.permute
- mindspore.ops.permute(input, axis)[source]
Permutes the dimensions of the input tensor according to input axis .
- Parameters
- Returns
Tensor, has the same dimension as input tensor, with axis suitably permuted.
- Raises
ValueError – If axis is None.
ValueError – If the number of elements of axis is not equal to input ndim.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> 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.]]]