Solved by James_Hickman in post #2 
              
              
                If anyone has a similar problem, the solution is quite simple, 
x=torch.tensor([[1,2],[3,4],[6,7],[8,9]]) 
print(torch.cat(torch.split(x,2),1)) 
tensor([[1, 2, 6, 7], [3, 4, 8, 9]])