mindspore.ops.moveaxis

查看源文件
mindspore.ops.moveaxis(x, source, destination)[源代码]

将输入tensor的两个轴调换位置。

更多参考详见 mindspore.ops.movedim()

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> x = mindspore.ops.zeros((3, 4, 5))
>>> output = mindspore.ops.moveaxis(x, 0, -1)
>>> print(output.shape)
(4, 5, 3)