I try to generate a distribution on gpu, but got segmentation fault.
Code is here:
from torch.distributions.multivariate_normal import MultivariateNormal
import torch
mean =  torch.ones(3).cuda()
scale = torch.ones(3).cuda()
mvn = MultivariateNormal(mean, torch.diag(scale))