mindspore.ops.swapdims

mindspore.ops.swapdims(input, dim0, dim1)

交换tensor的两个维度。 该函数和 mindspore.ops.swapaxes() 功能一致。

参数：

• input (Tensor) - 输入tensor。

• dim0 (int) - 第一个维度。

• dim1 (int) - 第二个维度。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> input = mindspore.tensor(mindspore.ops.ones([2, 3, 4]))
>>> output = mindspore.ops.swapdims(input, 0, 2)
>>> print(output.shape)
(4, 3, 2)