mindspore.mint.swapaxes

View Source On Gitee
mindspore.mint.swapaxes(input, axis0, axis1)[source]

Alias for mindspore.mint.transpose() . The input corresponds to the input in the reference interface, and the parameters axis0 and axis1 correspond to dim0 and dim1 in the reference interface respectively.

Warning

This is an experimental API that is subject to change or deletion.

Supported Platforms:

Ascend

Examples

>>> import numpy as np
>>> from mindspore import mint
>>> from mindspore import Tensor
>>> input = Tensor(np.ones((2,3,4), dtype=np.float32))
>>> output = mint.swapaxes(input, 0, 2)
>>> print(output.shape)
(4, 3, 2)