【torch】torch.roll

torch.roll

张量滚动

1 torch.roll(input, shifts, dims=None) → Tensor

input为输入张量
shift表示要滚动的幅度。负数表示左、上方向（二维时），正数表示右、下方向
dim表示在哪些维度上进行滚动。
如果dim给值，dim的len和shifts的len必须是一样的。比如dims=(a,b)），相当于在第a维度移x位，在b维度上移y位。