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 - 如果 axisstart 不是整数,或者 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)