Pt to onnx error:

RuntimeError: Exporting the operator uniform to ONNX opset version 12 is not supported. 
Please open a bug to request ONNX export support for the missing operator.

The reason is that during conversion, you can’t dynamically obtain the value in forward for operation, but complete the corresponding operation in init..