Solved by Brando_Miranda in post #11 
              
              
                check this out but summary use torch.stack if you want to respect the original nesting of the lists (by having a tensor with many indices and dimensions) and incrementally create it. There might be better ways but that works for me. 



# %%

import torch

# stack vs cat

# cat "extends" a list in t…