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位。