- 1. torch.onnx.export 参数
- 2. 自定义层处理
-
- 2.1 static 方式注册 - 通过Function方式的python方式进行自定义层定义
- bugs:
-
- 1. warnings
可以在构建pytorch的时候考虑自定义层能直接转onnx,可以实现带自定义层的pytorch —> onnx 一步到位
- 带参数和不带参数的自定义层构建
- 多个输出的自定义层构建
- 修剪onnx graph

可以在构建pytorch的时候考虑自定义层能直接转onnx,可以实现带自定义层的pytorch —> onnx 一步到位