mindspore.ops.movedim

mindspore.ops.movedim(x, source, destination)

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

参数：

• x (Tensor) - 输入tensor。

• source (Union[int, sequence[int]]) - 源维度。

• destination (Union[int, sequence[int]]) - 源维度的目标位置。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> # case1 : moving single axis
>>> import mindspore
>>> x = mindspore.tensor(mindspore.ops.zeros((3, 4, 5)))
>>> output = mindspore.ops.movedim(x, 0, -1)
>>> print(output.shape)
(4, 5, 3)
>>> # case 2 : moving multiple axes
>>> x = mindspore.tensor(mindspore.ops.zeros((3, 4, 5)))
>>> output = mindspore.ops.movedim(x, (0, 2), (1, 2))
>>> print(output.shape)
(4, 3, 5)