Inverse or torch.chunk (view version of torch.cat/stack)

[trancelestial (Bozidar Antic)]

Thank you, I will try implementing it that way!