Versions in this module Expand all Collapse all v1 v1.1.0 Jan 30, 2024 v1.0.0 Dec 14, 2023 Changes in this version + func ErrInvalidShape(expected onnx.Shape, actual []int) error + func ErrModel(format string, a ...any) error + func ModelProtoFromBytes(bytesModel []byte) (*onnx.ModelProto, error) + type InvalidShapeError struct + func (i InvalidShapeError) Error() string + type Model struct + GetOperator OpGetter + func NewModel(mp *onnx.ModelProto) (*Model, error) + func NewModelFromBytes(bytesModel []byte) (*Model, error) + func NewModelFromFile(path string) (*Model, error) + func NewModelFromZipFile(file *zip.File) (*Model, error) + func (m *Model) InputDimSize(input string, i int) (int, error) + func (m *Model) InputNames() []string + func (m *Model) InputShapes() onnx.Shapes + func (m *Model) OutputNames() []string + func (m *Model) OutputShape(output string) onnx.Shape + func (m *Model) OutputShapes() onnx.Shapes + func (m *Model) ParamNames() []string + func (m *Model) Run(inputs Tensors) (Tensors, error) + type OpGetter func(string) (ops.Operator, error) + func ResolveOperatorGetter(opsetID int64) (OpGetter, error) + type Tensors map[string]tensor.Tensor