torch.cat

>>> ones=torch.ones((2,3,2))
>>> zeros=torch.zeros((2,3,2))
>>> torch.cat((ones,zeros),1)
tensor([[[1., 1.],
         [1., 1.],
         [1., 1.],
         [0., 0.],
         [0., 0.],
         [0., 0.]],

        [[1., 1.],
         [1., 1.],
         [1., 1.],
         [0., 0.],
         [0., 0.],
         [0., 0.]]])
>>> torch.cat((ones,zeros),0)
tensor([[[1., 1.],
         [1., 1.],
         [1., 1.]],

        [[1., 1.],
         [1., 1.],
         [1., 1.]],

        [[0., 0.],
         [0., 0.],
         [0., 0.]],

        [[0., 0.],
         [0., 0.],
         [0., 0.]]])
>>> torch.cat((ones,zeros),-1)
tensor([[[1., 1., 0., 0.],
         [1., 1., 0., 0.],
         [1., 1., 0., 0.]],

        [[1., 1., 0., 0.],
         [1., 1., 0., 0.],
         [1., 1., 0., 0.]]])