pytorch 转换onnx_PyTorch VAE无法转换为onnx

I'm trying to convert a PyTorch VAE to onnx, but I'm getting: torch.onnx.symbolic.normal does not exist

The problem appears to originate from a reparametrize() function:

def reparametrize(self, mu, logvar):

std = logvar.mul(0.5).exp_()

if self.have_cuda:

eps = torch.normal(torch.zeros(std.size()),torch.ones(std.size())).cuda()

else:

eps = torch.normal(torch.zeros(std.size()),torch.ones(std.size()))

return eps.mul(std).add_(mu)

I also tried:

eps = torch.cuda.FloatTensor(std.size()).normal_()

which produced the error:

Schema not found for node. File a bug report.

Node: %173 : Float(1, 20) = aten::normal(%169, %170, %171, %172), scope: VAE

Input types:Float(1, 20), float, float, Generator

and

eps = torch.randn(std.size()).cuda()

<