2023年3月29日
算子开发
摘要： 算子开发 ## ONNX导出 ### torch.onnx.export 细解 `torch.onnx.export`中需要的模型实际上是一个`torch.jit.ScriptModule`。而要把普通 PyTorch 模型转一个这样的`TorchScript`模型，有跟踪（trace）和脚本化（script 阅读全文
posted @ 2023-03-29 16:16 nanmi 阅读(111) 评论(0) 推荐(0) 编辑