Hi Ameen!



 Ameen_Ali:

the first element (a) = Original.
the second element (b), b =  same as Original but the tensors are shifted by 1, meaning b[0 , : , :] = a[1 , : , : ]


Does torch.roll() 2 do part of what you want?
Best.
K. Frank