It wasn’t a typo, I indeed would like a tensor NxNx2F. Otherwise, cat would have been perfect  However, I’m not aware of something like this. I thought about something like torch.repeat but didn’t find exactly what I’m looking for