#off-by-one

An attempt at Tensor Puzzle 11 (reverse a vector of length i) is written as a[i - arange(i)]. What happens here?

def arange(i):
    return torch.tensor(range(i))

a = torch.tensor([10, 20, 30, 40])
out = a[4 - arange(4)]