Module onnx 

ONNX — Import/Export for interoperability with other frameworks.

Structs

OnnxGraph

A fully parsed ONNX graph: nodes + initializers + I/O metadata.

OnnxModel

An ONNX model with graph, metadata, and opset information.

OnnxNode

An ONNX graph node (operation).

OnnxTensor

Represents an ONNX TensorProto (a named tensor with shape and data).

Enums

OnnxAttribute

An ONNX attribute value.

Functions

export_tensors

Export named tensors directly to ONNX format.

export_weights

Export a module’s weights as an ONNX model file.

load_onnx_graph

Load a full ONNX graph (nodes + initializers) from a file.

load_onnx_graph_from_bytes

Load a full ONNX graph from in-memory bytes.

load_onnx_weights

Load tensor weights from an ONNX model file.

load_onnx_weights_from_bytes

Load tensor weights from ONNX bytes (in-memory).

run_onnx_graph

Execute an ONNX graph on the given backend.