torch.roll

原理：将tensor进行偏移。例如[1,2,3]变成[2,3,1]，这个就是roll。又再如[1,2,3]变成[3,1,2]。前者向左偏移一个单位，后者向左偏移两个单位。

明白了原理就很简单了。

torch.roll(input, shifts, dims=None)

import torch
x = torch.tensor([[1, 2],[ 3,4]])

print(x)
torch.roll(x,1,-1)#表示最后一个维度上进行操作，并且向右偏移1个单位。

print(x)
torch.roll(x,-1,0)#表示在0维度上进行操作，并且向上偏移1个单位。

print(x)
torch.roll(x,(2,2),(0,1))#表示两个维度上都要进行偏移，并且各自在正方向上偏移两个单位。