【Pytorch小知识】torch.roll()函数的用法及在Swin Transformer中的应用(详细易懂)

---

torch.roll()的官方文档

函数形式及参数说明

函数形式：torch.roll(input,shifts,dims=None）
参数解释：

• input:输入张量
• shifts:滚动的方向和长度，若为正，则向下滚动；若为负，则向上滚动。可是一个整数也可以是一个元组。(具体见下面例子)
• dims:张量按什么维度滚动。以二维为例，0就是按行上下或者下上(shifts为负)滚动；1就是左右或者右左(shifts为负)。可以是整数或元组。