mindspore.ops.swapdims

View Source On Gitee
mindspore.ops.swapdims(input, dim0, dim1)[source]

Interchange two dims of a tensor. This function is equivalent to mindspore.ops.swapaxes() function.

Parameters

  • input (Tensor) – The input tensor.

  • dim0 (int) – First dim.

  • dim1 (int) – Second dim.

Returns

Tensor

Supported Platforms:

Ascend GPU CPU

Examples

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