Pytorch Cat Function