Model

class onnx_ir.Model(graph, *, ir_version, producer_name=None, producer_version=None, domain=None, model_version=None, doc_string=None, functions=(), metadata_props=None)

参数:

• graph (Graph)

• ir_version (int)

• producer_name (str | None)

• producer_version (str | None)

• domain (str | None)

• model_version (int | None)

• doc_string (str | None)

• functions (Sequence[Function])

• metadata_props (dict[str, str] | None)

doc_string

domain

graph

ir_version

model_version

producer_name

producer_version

display(*, page=False)

漂亮地打印对象。

参数:

page (bool) – 是否分页输出。

返回类型:

无

property functions: dict[tuple[str, str, str], Function]

graphs()

获取模型中的所有图和子图。

这是一个遍历模型的便捷方法。对于更高级的节点遍历，请考虑使用 onnx_ir.traversal.RecursiveGraphIterator。

返回类型:

Iterable[Graph]

property meta: MetadataStore

用于中间分析的元数据存储。

如果您希望将元数据序列化到 ONNX proto，请写入 metadata_props。

property metadata_props: dict[str, str]

模型的元数据属性。

元数据属性用于存储模型的附加信息。与 meta 不同，此属性会序列化到 ONNX proto。

property opset_imports: dict[str, int]