Module model

Module model 

Source
Expand description

Model operations for ONNX compatibility

This module defines operations for training, saving, and loading models compatible with ONNX format.

Structs§

ModelMetadata
ONNX Model metadata
OnnxGraph
ONNX Graph representation
OnnxModel
ONNX Model representation
OnnxNode
ONNX Node representation
OnnxTensor
ONNX Tensor representation
TrainingState
Training state for a model

Enums§

LossKind
Loss function types
ModelKind
Model save/load operations
OnnxAttribute
ONNX Attribute value
OptimizerKind
Optimizer types
TrainingKind
Training-related operations