Pytorch Horizontal Stack