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 - dim0 或 dim1 不是整数。
ValueError - dim0 或 dim1 不在 \([-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)