（beta）torch_npu.contrib.function.roll

产品支持情况

[object Object]undefined

功能说明

使用NPU亲和写法替换swin-transformer中的原生roll。

函数原型

参数说明

• input1 ([object Object])：输入张量。
• shifts ([object Object] of [object Object])：每个维度张量滚动（roll）的位移量。
• dims ([object Object] of [object Object])：要滚动的维度。

返回值说明

[object Object]

滚动之后的结果。

约束说明

[object Object]是4维张量，[object Object]和[object Object]的长度为2。

调用示例