19 lines
683 B
ReStructuredText
19 lines
683 B
ReStructuredText
mindspore.ops.Roll
|
||
===================
|
||
|
||
.. py:class:: mindspore.ops.Roll(shifts, dims=None)
|
||
|
||
沿轴移动Tensor的元素。
|
||
|
||
更多参考详见 :func:`mindspore.ops.roll`。
|
||
|
||
参数：
|
||
- **shifts** (Union[list(int), tuple(int), int]) - 指定元素移动方式，如果为正数，表示元素沿指定维度正向移动（朝向较大的索引）的位置数。负偏移将向相反的方向滚动元素。
|
||
- **dims** (Union[list(int), tuple(int), int]) - 指定需移动维度的轴。默认值： ``None`` 。
|
||
|
||
输入：
|
||
- **input** (Tensor) - 输入Tensor。
|
||
|
||
输出：
|
||
Tensor，shape和数据类型与输入 `input` 相同。
|