Versions in this module Expand all Collapse all v0 v0.3.0 Aug 25, 2025 Changes in this version + func UnregisterLayer(opType string) type Model + ZMFVersion string + func LoadModelFromZMF[T tensor.Numeric](engine compute.Engine[T], ops numeric.Arithmetic[T], filePath string) (*Model[T], error) v0.2.0 Aug 22, 2025 Changes in this version + func BuildFromZMF[T tensor.Numeric](engine compute.Engine[T], ops numeric.Arithmetic[T], model *zmf.Model) (*graph.Graph[T], error) + func DecodeTensor[T tensor.Numeric](tensorProto *zmf.Tensor) (*tensor.TensorNumeric[T], error) + func EncodeTensor[T tensor.Numeric](t *tensor.TensorNumeric[T]) (*zmf.Tensor, error) + func LoadZMF(filePath string) (*zmf.Model, error) + func RegisterLayer[T tensor.Numeric](opType string, builder LayerBuilder[T]) + type Exporter interface + Export func(model *Model[T], path string) error + type LayerBuilder func(engine compute.Engine[T], ops numeric.Arithmetic[T], name string, ...) (graph.Node[T], error) + func GetLayerBuilder[T tensor.Numeric](opType string) (LayerBuilder[T], error) + type Model struct + Embedding *embeddings.TokenEmbedding[T] + Graph *graph.Graph[T] + func NewModel[T tensor.Numeric](embedding *embeddings.TokenEmbedding[T], graph *graph.Graph[T]) *Model[T] + func (m *Model[T]) Forward(ctx context.Context, inputs *tensor.TensorNumeric[int]) (*tensor.TensorNumeric[T], error)
Go to main page