使用NPU亲和写法替换swin-transformer中的原生roll。
[object Object]
- 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。
[object Object]