[pytorch] torch.onnx.export 使用及自定义层构建

文章目录

  • 1. torch.onnx.export 参数
  • 2. 自定义层处理
    • 2.1 static 方式注册 - 通过Function方式的python方式进行自定义层定义
  • bugs:
    • 1. warnings

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

  • 带参数和不带参数的自定义层构建
  • 多个输出的自定义层构建
  • 修剪onnx graph

1. torch.onnx.export 参数

 
 

你可能感兴趣的:(模型部署,深度学习,pytorch,深度学习,onnx)