文件名称：onnx-simplifier:简化您的Onnx模型
文件大小：151KB
文件格式：ZIP
更新时间：2024-05-18 21:31:33
Python
ONNX简化器 ONNX很棒，但有时太复杂了。 背景 有一天，我想将以下简单的重塑操作导出到ONNX： import torch class JustReshape ( torch . nn . Module ): def __init__ ( self ): super ( JustReshape , self ). __init__ () def forward ( self , x ): return x . view (( x . shape [ 0 ], x . shape [ 1 ], x . shape [ 3 ], x . shape [ 2 ])) net = JustReshape () model_name = 'just_reshape.onnx' dummy_input = torch . randn ( 2 ,
【文件预览】：
onnx-simplifier-master
----setup.py(1KB)
----.gitignore(2KB)
----.travis.yml(358B)
----LICENSE(11KB)
----.github()
--------ISSUE_TEMPLATE()
----onnxsim()
--------__init__.py(68B)
--------onnx_simplifier.py(19KB)
--------__main__.py(5KB)
----README.md(3KB)
----imgs()
--------complicated_reshape.png(26KB)
--------simple_reshape.png(5KB)
--------comparison.png(111KB)