I think you highlight a good point ! Maybe I’m asking something which is not possible the way I explained it.
I give an example, it would be useful:
a = torch.FloatTensor([[1,2,3],[4,5,6]])
b = torch.FloatTensor([[10,20,30],[40,50,60]])

I would like as ouput something like: (final dimension NxNx2F. Here for simplicity it’s simply N*Nx2F)
1 2 3 1 2 3
1 2 3 4 5 6
1 2 3 10 20 30
1 2 3 40 50 60
4 5 6 1 2 3
4 5 6 4 5 6
4 5 6 10 20 30
4 5 6 40 50 60
10 20 30 1 2 3
10 20 30 4 5 6
10 20 30 10 20 30
10 20 30 40 50 60
40 50 60 1 2 3
40 50 60 4 5 6
40 50 60 10 20 30
40 50 60 40 50 60

Does it make more sense now ?