Pytorch Cat Function In R