mindspore.ops.swapaxes
- mindspore.ops.swapaxes(input, axis0, axis1)[source]
Interchange two axes of a tensor.
- Parameters
- Returns
Transposed tensor, has the same data type as input.
- Raises
TypeError – If argument input is not Tensor.
TypeError – If axis0 or axis1 is not integer.
ValueError – If axis0 or axis1 is not in the range of \([-ndim, ndim-1]\).
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import numpy as np >>> from mindspore import ops >>> from mindspore import Tensor >>> input = Tensor(np.ones((2,3,4), dtype=np.float32)) >>> output = ops.swapaxes(input, 0, 2) >>> print(output.shape) (4, 3, 2)