mindspore.ops.swapdims

查看源文件
mindspore.ops.swapdims(input, dim0, dim1)[源代码]

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

参数:
  • input (Tensor) - 输入Tensor。

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

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

返回:

转化后的Tensor,与输入具有相同的数据类型。

异常:
  • TypeError - input 不是Tensor类型。

  • TypeError - dim0dim1 不是整数。

  • ValueError - dim0dim1 不在 \([-ndim, ndim-1]\) 范围内。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import ops
>>> from mindspore import Tensor
>>> input = Tensor(np.ones((2,3,4), dtype=np.float32))
>>> output = ops.swapdims(input, 0, 2)
>>> print(output.shape)
(4, 3, 2)