Suppose we have an array:
A = torch.arange(start = 1, end = 101)
And we want to take a rolling window over the array to produce matrix B such that:
B=([1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
      [2, 3, 4, ...],
      [3, 4, 5, ...],
      [...],
      [91, 92, 93, 94, ..., 99, 100])

I know we could run a loop through to make it, but is there a faster, parallelized way in Pytorch to render this?