hello, Given a tensor of shape Original = [A , B , C] I want to build a list of tensors (target) in the following way: 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 , : , : ] the third element ©, c = …

Hi Ameen! [image] 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() do part of what you want? Best. K. Frank

Iterate over all tensor permuation in speceific dim

KFrank (K. Frank) March 5, 2020, 5:42pm 2

Hi Ameen!

Does torch.roll() do part of what you want?

Best.

K. Frank

1 Like