onnx_ir

函数和构造函数

onnx_ir.load

从文件中加载 ONNX 模型。

onnx_ir.save

将 ONNX 模型保存到文件。

onnx_ir.from_proto

将 ONNX proto 消息反序列化为 IR 对象。

onnx_ir.from_onnx_text

将 ONNX 文本表示转换为 IR 模型。

onnx_ir.to_proto

将 IR 对象序列化为 proto。

onnx_ir.to_onnx_text

将 IR 模型转换为 ONNX 文本表示。

onnx_ir.tensor

从 ArrayLike 对象或 TensorProto 创建张量值。

onnx_ir.val

创建具有给定名称和类型的Value

onnx_ir.node

创建Node

onnx_ir.TensorProtocol

由数据支持的具体张量。

onnx_ir.Value

IR 值。

onnx_ir.Node

IR 节点。

onnx_ir.Graph

IR 图。

onnx_ir.Model

onnx_ir.GraphView

图的只读视图。

onnx_ir.Function

IR 函数。

onnx_ir.Attr

ONNX 属性或引用的基类。

onnx_ir.Shape

表示张量的形状,包括其维度和可选的标注。

onnx_ir.SymbolicDim

不可变的符号维度,可以在多个形状之间共享。

onnx_ir.TypeAndShape

类型和形状。

onnx_ir.TensorType

表示张量的类型。

onnx_ir.SparseTensorType

表示稀疏张量的类型。

onnx_ir.SequenceType

表示元素序列的类型。

onnx_ir.OptionalType

表示可选元素的类型。

onnx_ir.Tensor

不可变的具体张量。

onnx_ir.ExternalTensor

一个不可变的具体张量,其数据存储在磁盘上。

onnx_ir.StringTensor

多维字符串数组(作为二进制数据以匹配 TensorProto 中的 string_data 字段)。

onnx_ir.LazyTensor

一个惰性评估函数以获取实际张量的张量。

onnx_ir.PackedTensor

以打包格式存储 4 位数据类型的张量。

枚举

onnx_ir.DataType

ONNX 张量数据类型的枚举,定义在onnx.TensorProto中。

onnx_ir.AttributeType

ONNX 属性类型的枚举。

内部容器

onnx_ir._graph_containers.GraphInitializers

Graph 的初始化器,表示为dict[str, Value],并附带额外的变异方法。

onnx_ir._graph_containers.Attributes

Node 的属性,表示为dict[str, Attr],并附带额外的访问方法。

onnx_ir._metadata.MetadataStore

用于存储 IR 对象元数据的类。