import torcha = torch.ones([1,2])b = torch.ones([1,2])torch.stack([a,b],1)(0 ,.,.) =  1 1 1 1[torch.FloatTensor of size 1x2x2]