mindspore.mint.swapaxes

查看源文件
mindspore.mint.swapaxes(input, axis0, axis1)[源代码]

警告

这是一个实验性API,后续可能修改或删除。

mindspore.mint.transpose() 的别名。

更多参考详见 mindspore.mint.transpose()

支持平台:

Ascend

样例:

>>> 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)