mindspore.ops.swapaxes
- mindspore.ops.swapaxes(input, axis0, axis1)[源代码]
交换Tensor的两个维度。
- 参数:
input (Tensor) - 输入Tensor。
axis0 (int) - 第一个维度。
axis1 (int) - 第二个维度。
- 返回:
转化后的Tensor,与输入具有相同的数据类型。
- 异常:
TypeError - input 不是Tensor类型。
TypeError - axis0 或 axis1 不是整数。
ValueError - axis0 或 axis1 不在 \([-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.swapaxes(input, 0, 2) >>> print(output.shape) (4, 3, 2)