mindspore.ops.roll

查看源文件
mindspore.ops.roll(input, shifts, dims=None)[源代码]

沿轴移动Tensor的元素。

参数:
  • input (Tensor) - 输入Tensor。

  • shifts (Union[list(int), tuple(int), int]) - 指定元素移动方式,如果为正数,则元素沿指定维度正向移动(朝向较大的索引)的位置数。负偏移将向相反的方向滚动元素。

  • dims (Union[list(int), tuple(int), int], 可选) - 指定需移动维度的轴。默认值: None 。如果dims为 None ,则会将输入Tensor展平后再进行roll计算,然后将计算结果reshape为输入的shape。

返回:

Tensor,shape和数据类型与输入 input 相同。

异常:
  • TypeError - shifts 不是int、tuple或list。

  • TypeError - dims 不是int、tuple或list。

支持平台:

GPU

样例:

>>> import numpy as np
>>> import mindspore as ms
>>> from mindspore import ops
>>> from mindspore import Tensor
>>> input = Tensor(np.array([0, 1, 2, 3, 4]).astype(np.float32))
>>> output = ops.roll(input, shifts=2, dims=0)
>>> print(output)
[3. 4. 0. 1. 2.]