Gaussian Mixture Model Pytorch Cuda