（beta）torch_npu.contrib.function.roll

函数原型

功能说明

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

参数说明

• input1 (Tensor) - 输入张量。
• shifts (Int或Tuple of python:ints) - 张量元素移动的位置数。如果该shift组成的是tuple，则dims必须是大小相同的tuple，每个维度都将按相应的值roll。
• dims (Int或Tuple of python:ints) - 沿着roll的轴。

输出说明

Tensor - shifted input。

支持的型号

• [object Object]Atlas 训练系列产品[object Object]
• [object Object]Atlas A2 训练系列产品[object Object]
• [object Object]Atlas A3 训练系列产品[object Object]
• [object Object]Atlas 推理系列产品[object Object]

调用示例