mindspore.numpy.rollaxis
- mindspore.numpy.rollaxis(x, axis, start=0)[源代码]
将指定轴向后滚动,直到它位于给定的位置。 其他轴的相对位置保持不变。
- 参数:
x (Tensor) - 要进行调换的Tensor。
axis (int) - 要滚动的轴。
start (int) - 默认值:
0
。如果 \(start \leq axis\) ,则将轴向后滚动,直到它位于这个位置(start
)。如果 \(start > axis\) ,则将轴滚动直到它位于此位置之前(start
)。如果 \(start < 0\) ,则start
会被归一化为非负数(详细信息见源码)。
- 返回:
转置后的Tensor,数据类型与原Tensor x 相同。
- 异常:
TypeError - 如果
axis
或start
不是整数,或者x
不是Tensor。ValueError - 如果
axis
不在 \([-ndim, ndim-1]\) 范围内,或者start
不在 \([-ndim, ndim]\) 范围内。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore.numpy as np >>> x = np.ones((2,3,4)) >>> output = np.rollaxis(x, 0, 2) >>> print(output.shape) (3, 2, 4)