[解决] We don‘t have an op for aten::normal but it isn‘t a special case

pytorch 网络转为 onnx时遇到的问题：

错误

The error is :
RuntimeError: 0 INTERNAL ASSERT FAILED at “/pytorch/torch/csrc/jit/ir/alias_analysis.cpp”:532, please report a bug to PyTorch. We don’t have an op for aten::normal but it isn’t a special case. Argument types: Tensor, float, float, None

意思：
运行时错误：0 内部断言失败在“/pytorch/torch/csrc/jit/ir/alias_analysis.cpp”：532，请向 PyTorch 报告错误。没有 aten::normal 的操作，但这不是一个特例。参数类型：张量、浮点数、浮点数、None

系统环境

OS: Ubuntu 18.04.6 LTS
Python version: 3.7 (64-bit runtime)
Is CUDA available: True
CUDA runtime version: 11.1.74
GPU models and configuration: GPU 0: NVIDIA GeForce RTX 3090 Ti
Nvidia driver version: 510.108.03

出错原因

原始代码中需要生成一个和输入相同的均值为0，方差为1的标准正态分布，使用了normal_()

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

替换为下面这句，能达到同样的功能

eps = torch.randn_like(std)

可以成功转换