2022-10-27 15:37:57 +08:00
|
|
|
|
mindspore.ops.roll
|
|
|
|
|
|
===================
|
|
|
|
|
|
|
2022-11-03 10:28:05 +08:00
|
|
|
|
.. py:function:: mindspore.ops.roll(x, shifts, dims=None)
|
2022-10-27 15:37:57 +08:00
|
|
|
|
|
|
|
|
|
|
沿轴移动Tensor的元素。
|
|
|
|
|
|
|
|
|
|
|
|
参数：
|
|
|
|
|
|
- **x** (Tensor) - 输入Tensor。
|
|
|
|
|
|
- **shifts** (Union[list(int), tuple(int), int]) - 指定元素移动方式，如果为正数，则元素沿指定维度正向移动（朝向较大的索引）的位置数。负偏移将向相反的方向滚动元素。
|
2023-01-16 16:58:53 +08:00
|
|
|
|
- **dims** (Union[list(int), tuple(int), int], optional) - 指定需移动维度的轴。默认值：None。如果dims为None，则会将输入Tensor展平后再进行roll计算，然后将计算结果reshape为输入的shape。
|
2022-10-27 15:37:57 +08:00
|
|
|
|
|
|
|
|
|
|
返回：
|
|
|
|
|
|
Tensor，shape和数据类型与输入 `x` 相同。
|
|
|
|
|
|
|
|
|
|
|
|
异常：
|
|
|
|
|
|
- **TypeError** - `shifts` 不是int、tuple或list。
|
|
|
|
|
|
- **TypeError** - `dims` 不是int、tuple或list。
|
