什么库支持torch.onnx.export
时间: 2024-01-03 13:55:45 浏览: 145
nnunet-pytorch转onnx.zip
5星 · 资源好评率100%
支持torch.onnx.export的库是PyTorch。这个库提供了torch.onnx.export函数,用于将PyTorch模型导出为ONNX格式。该函数需要传入三个参数:mode(torch.nn.Module、torch.jit.ScriptModule或torch.jit.ScriptFunction)、需要转换的模型和args(元组或torch.Tensor)。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [torch.onnx.export详细介绍](https://blog.csdn.net/jiong9412/article/details/125383053)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
阅读全文