mindspore.numpy.moveaxis
- mindspore.numpy.moveaxis(a, source, destination)[源代码]
将数组的轴移动到新位置。 其他轴保持原始顺序不变。
- 参数:
a (Tensor) - 原数组,返回数组的shape和数据类型与
a
相同。待修改source (int, ints的序列) - 要移动的轴的原始位置。这些位置必须唯一。
destination (int, ints的序列) - 每个原始轴的新位置。这些位置也必须唯一。
- 返回:
Tensor,已经移动过轴的数组。
- 异常:
ValueError - 如果轴超出范围 \([-a.ndim, a.ndim)\) ,或者轴中包含重复项。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore.numpy as np >>> x = np.zeros((3, 4, 5)) >>> output = np.moveaxis(x, 0, -1) >>> print(output.shape) (4, 5, 3) >>> output = np.moveaxis(x, -1, 0) >>> print(output.shape) (5, 3, 4) >>> output = np.moveaxis(x, [0, 1, 2], [-1, -2, -3]) >>> print(output.shape) (5, 4, 3)