mindspore.mint.swapaxes
- mindspore.mint.swapaxes(input, axis0, axis1)[源代码]
mindspore.mint.transpose()
的别名。其中参数 input 对应参考接口的 input ,参数 axis0 和 axis1 分别对应参考接口的 dim0 和 dim1 。更多参考详见
mindspore.mint.transpose()
。警告
这是一个实验性API,后续可能修改或删除。
- 支持平台:
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)