导出

onnx_graphsurgeon.export_onnx(graph: Graph, do_type_check=True, **kwargs) → ModelProto

将 onnx-graphsurgeon Graph 导出到 ONNX 模型。

参数:

  • graph (Graph) – 要导出的图

  • do_type_check (bool) – 是否检查输入和输出张量是否定义了数据类型，如果没有则失败。默认为 True。

  • kwargs – 传递给 onnx.helper.make_model 的其他参数

返回:

对应的 ONNX 模型。

返回类型:

onnx.ModelProto