Oh, but i thought that
torch.cat() or torch.stack()
would change the dimensions, from an example from Stackoverflow
So if  A  and  B  are of shape (3, 4),  torch.cat([A, B], dim=0)  will be of shape (6, 4) and  torch.stack([A, B], dim=0)  will be of shape (2, 3, 4).
Which in my case would be
torch.cat([A,B], dim=1) resulting in [12 x 14 x 368 x 640]?
Does this still work when I want the output to be [12 x 7 x 368 x 640]?