Why Go
- Case Studies
  
  Common problems companies solve with Go
- Use Cases
  
  Stories about how and why companies use Go
- Security
  
  How Go can help keep you secure by default
Learn
Docs
- Effective Go
  
  Tips for writing clear, performant, and idiomatic Go code
- Go User Manual
  
  A complete introduction to building software with Go
- Standard library
  
  Reference documentation for Go's standard library
- Release Notes
  
  Learn what's new in each Go release
Packages
Community
- Recorded Talks
  
  Videos from prior events
- Meetups
  
  Meet other local Go developers
- Conferences
  
  Learn and network with Go developers from around the world
- Go blog
  
  The Go project's official blog.
- Go project
  
  Get help and stay informed from Go
- Get connected

core

package

Go to main page

Versions in this module

v0

v0.3.0

Aug 25, 2025

Changes in this version

+ func BuildCast[T tensor.Numeric](engine compute.Engine[T], _ numeric.Arithmetic[T], _ string, ...) (graph.Node[T], error)

+ func BuildConcat[T tensor.Numeric](engine compute.Engine[T], _ numeric.Arithmetic[T], _ string, ...) (graph.Node[T], error)

+ func BuildMatMul[T tensor.Numeric](engine compute.Engine[T], _ numeric.Arithmetic[T], _ string, ...) (graph.Node[T], error)

+ func BuildMul[T tensor.Numeric](engine compute.Engine[T], _ numeric.Arithmetic[T], _ string, ...) (graph.Node[T], error)

+ func BuildReshape[T tensor.Numeric](engine compute.Engine[T], _ numeric.Arithmetic[T], _ string, ...) (graph.Node[T], error)

+ func BuildRotaryEmbedding[T tensor.Numeric](engine compute.Engine[T], _ numeric.Arithmetic[T], name string, ...) (graph.Node[T], error)

+ func BuildShape[T tensor.Numeric](engine compute.Engine[T], _ numeric.Arithmetic[T], _ string, ...) (graph.Node[T], error)

+ func BuildSpectralFingerprint[T tensor.Numeric](engine compute.Engine[T], ops numeric.Arithmetic[T], _ string, ...) (graph.Node[T], error)

+ func BuildSub[T tensor.Numeric](engine compute.Engine[T], _ numeric.Arithmetic[T], _ string, ...) (graph.Node[T], error)

+ func BuildUnsqueeze[T tensor.Numeric](engine compute.Engine[T], _ numeric.Arithmetic[T], _ string, ...) (graph.Node[T], error)

+ type Add struct

+ func NewAdd[T tensor.Numeric](engine compute.Engine[T]) *Add[T]

+ func (a *Add[T]) Attributes() map[string]any

+ func (a *Add[T]) Backward(_ context.Context, mode types.BackwardMode, dOut *tensor.TensorNumeric[T], ...) ([]*tensor.TensorNumeric[T], error)

+ func (a *Add[T]) Forward(ctx context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (a *Add[T]) OpType() string

+ func (a *Add[T]) OutputShape() []int

+ func (a *Add[T]) Parameters() []*graph.Parameter[T]

type Bias

+ func NewBiasFromParam[T tensor.Numeric](engine compute.Engine[T], ops numeric.Arithmetic[T], ...) *Bias[T]

+ func (b *Bias[T]) Attributes() map[string]interface{}

+ func (b *Bias[T]) OpType() string

+ type BiasOption func(*BiasOptions[T])

+ func WithBiasInitializer[T tensor.Numeric](initializer func(size int) []T) BiasOption[T]

+ type BiasOptions struct

+ Initializer func(size int) []T

+ type Cast struct

+ func NewCast[T tensor.Numeric](engine compute.Engine[T]) *Cast[T]

+ func (c *Cast[T]) Attributes() map[string]interface{}

+ func (c *Cast[T]) Backward(_ context.Context, mode types.BackwardMode, ...) ([]*tensor.TensorNumeric[T], error)

+ func (c *Cast[T]) Forward(_ context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (c *Cast[T]) OpType() string

+ func (c *Cast[T]) OutputShape() []int

+ func (c *Cast[T]) Parameters() []*graph.Parameter[T]

+ type Concat struct

+ func NewConcat[T tensor.Numeric](engine compute.Engine[T], axis int) *Concat[T]

+ func (c *Concat[T]) Attributes() map[string]interface{}

+ func (c *Concat[T]) Backward(_ context.Context, mode types.BackwardMode, ...) ([]*tensor.TensorNumeric[T], error)

+ func (c *Concat[T]) Forward(_ context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (c *Concat[T]) OpType() string

+ func (c *Concat[T]) OutputShape() []int

+ func (c *Concat[T]) Parameters() []*graph.Parameter[T]

type Dense

+ func NewDenseFromParams[T tensor.Numeric](linear *Linear[T], bias *Bias[T]) *Dense[T]

+ func (d *Dense[T]) Attributes() map[string]interface{}

+ func (d *Dense[T]) OpType() string

+ type DenseOption func(*DenseOptions[T])

+ func WithBias[T tensor.Numeric](withBias bool) DenseOption[T]

+ type DenseOptions struct

+ WithBias bool

+ type FFN struct

+ func NewFFN[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], ...) (*FFN[T], error)

+ func (f *FFN[T]) Attributes() map[string]interface{}

+ func (f *FFN[T]) Backward(ctx context.Context, mode types.BackwardMode, dOut *tensor.TensorNumeric[T], ...) ([]*tensor.TensorNumeric[T], error)

+ func (f *FFN[T]) Engine() compute.Engine[T]

+ func (f *FFN[T]) Forward(ctx context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (f *FFN[T]) OpType() string

+ func (f *FFN[T]) OutputShape() []int

+ func (f *FFN[T]) Parameters() []*graph.Parameter[T]

+ type FFNOption func(*FFNOptions[T])

+ func WithFFNBias[T tensor.Numeric](withBias bool) FFNOption[T]

+ func WithFFNInitializer[T tensor.Numeric](initializer components.WeightInitializer[T]) FFNOption[T]

+ type FFNOptions struct

+ Initializer components.WeightInitializer[T]

+ WithBias bool

+ type FiLM struct

+ func NewFiLM[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], ...) (*FiLM[T], error)

+ func (f *FiLM[T]) Attributes() map[string]interface{}

+ func (f *FiLM[T]) Backward(ctx context.Context, mode types.BackwardMode, ...) ([]*tensor.TensorNumeric[T], error)

+ func (f *FiLM[T]) Forward(ctx context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (f *FiLM[T]) OpType() string

+ func (f *FiLM[T]) OutputShape() []int

+ func (f *FiLM[T]) Parameters() []*graph.Parameter[T]

+ type LMHead struct

+ func NewLMHead[T tensor.Numeric](engine compute.Engine[T], ops numeric.Arithmetic[T], hiddenDim, vocabSize int) (*LMHead[T], error)

+ func (h *LMHead[T]) Backward(ctx context.Context, mode types.BackwardMode, dOut *tensor.TensorNumeric[T], ...) ([]*tensor.TensorNumeric[T], error)

+ func (h *LMHead[T]) Forward(ctx context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (h *LMHead[T]) OutputShape() []int

+ func (h *LMHead[T]) Parameters() []*graph.Parameter[T]

+ func (h *LMHead[T]) SetWeights(weights *tensor.TensorNumeric[T])

type Linear

+ func NewLinearFromParam[T tensor.Numeric](engine compute.Engine[T], weights *graph.Parameter[T]) *Linear[T]

+ func (l *Linear[T]) Attributes() map[string]interface{}

+ func (l *Linear[T]) OpType() string

+ type LinearOption func(*LinearOptions[T])

+ func WithHe[T tensor.Numeric](ops numeric.Arithmetic[T]) LinearOption[T]

+ func WithInitializer[T tensor.Numeric](initializer components.WeightInitializer[T]) LinearOption[T]

+ func WithUniform[T tensor.Numeric](ops numeric.Arithmetic[T], scale float64) LinearOption[T]

+ func WithXavier[T tensor.Numeric](ops numeric.Arithmetic[T]) LinearOption[T]

+ type LinearOptions struct

+ Initializer components.WeightInitializer[T]

+ type MatMul struct

+ func NewMatMul[T tensor.Numeric](engine compute.Engine[T]) *MatMul[T]

+ func (m *MatMul[T]) Attributes() map[string]interface{}

+ func (m *MatMul[T]) Backward(ctx context.Context, mode types.BackwardMode, ...) ([]*tensor.TensorNumeric[T], error)

+ func (m *MatMul[T]) Forward(ctx context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (m *MatMul[T]) OpType() string

+ func (m *MatMul[T]) OutputShape() []int

+ func (m *MatMul[T]) Parameters() []*graph.Parameter[T]

+ type Mul struct

+ func NewMul[T tensor.Numeric](engine compute.Engine[T]) *Mul[T]

+ func (m *Mul[T]) Attributes() map[string]interface{}

+ func (m *Mul[T]) Backward(ctx context.Context, mode types.BackwardMode, ...) ([]*tensor.TensorNumeric[T], error)

+ func (m *Mul[T]) Forward(ctx context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (m *Mul[T]) OpType() string

+ func (m *Mul[T]) OutputShape() []int

+ func (m *Mul[T]) Parameters() []*graph.Parameter[T]

type PolynomialExpansion

+ func (p *PolynomialExpansion[T]) Attributes() map[string]interface{}

+ func (p *PolynomialExpansion[T]) OpType() string

+ type PolynomialExpansionOption func(*PolynomialExpansionOptions[T])

+ func WithPolynomialBias[T tensor.Numeric](includeBias bool) PolynomialExpansionOption[T]

+ func WithPolynomialDegree[T tensor.Numeric](degree int) PolynomialExpansionOption[T]

+ type PolynomialExpansionOptions struct

+ Degree int

+ IncludeBias bool

+ type Reshape struct

+ func NewReshape[T tensor.Numeric](engine compute.Engine[T], targetShape []int) *Reshape[T]

+ func (r *Reshape[T]) Attributes() map[string]interface{}

+ func (r *Reshape[T]) Backward(ctx context.Context, mode types.BackwardMode, ...) ([]*tensor.TensorNumeric[T], error)

+ func (r *Reshape[T]) Forward(ctx context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (r *Reshape[T]) OpType() string

+ func (r *Reshape[T]) OutputShape() []int

+ func (r *Reshape[T]) Parameters() []*graph.Parameter[T]

+ type RotaryEmbedding struct

+ func NewRotaryEmbedding[T tensor.Numeric](engine compute.Engine[T]) *RotaryEmbedding[T]

+ func (r *RotaryEmbedding[T]) Attributes() map[string]interface{}

+ func (r *RotaryEmbedding[T]) Backward(_ context.Context, mode types.BackwardMode, ...) ([]*tensor.TensorNumeric[T], error)

+ func (r *RotaryEmbedding[T]) Forward(_ context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (r *RotaryEmbedding[T]) Name() string

+ func (r *RotaryEmbedding[T]) OpType() string

+ func (r *RotaryEmbedding[T]) OutputShape() []int

+ func (r *RotaryEmbedding[T]) Parameters() []*graph.Parameter[T]

+ func (r *RotaryEmbedding[T]) SetName(name string)

+ type Shape struct

+ func New[T tensor.Numeric](engine compute.Engine[T]) *Shape[T]

+ func (s *Shape[T]) Attributes() map[string]interface{}

+ func (s *Shape[T]) Backward(_ context.Context, mode types.BackwardMode, _ *tensor.TensorNumeric[T], ...) ([]*tensor.TensorNumeric[T], error)

+ func (s *Shape[T]) Forward(_ context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (s *Shape[T]) OpType() string

+ func (s *Shape[T]) OutputShape() []int

+ func (s *Shape[T]) Parameters() []*graph.Parameter[T]

+ type SpectralFingerprint struct

+ func NewSpectralFingerprint[T tensor.Numeric](engine compute.Engine[T], ops numeric.Arithmetic[T], window, topK int) (*SpectralFingerprint[T], error)

+ func (s *SpectralFingerprint[T]) Attributes() map[string]interface{}

+ func (s *SpectralFingerprint[T]) Backward(_ context.Context, mode types.BackwardMode, _ *tensor.TensorNumeric[T], ...) ([]*tensor.TensorNumeric[T], error)

+ func (s *SpectralFingerprint[T]) Forward(_ context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (s *SpectralFingerprint[T]) OpType() string

+ func (s *SpectralFingerprint[T]) OutputShape() []int

+ func (s *SpectralFingerprint[T]) Parameters() []*graph.Parameter[T]

+ type Sub struct

+ func NewSub[T tensor.Numeric](engine compute.Engine[T]) *Sub[T]

+ func (s *Sub[T]) Attributes() map[string]interface{}

+ func (s *Sub[T]) Backward(ctx context.Context, mode types.BackwardMode, ...) ([]*tensor.TensorNumeric[T], error)

+ func (s *Sub[T]) Forward(ctx context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (s *Sub[T]) OpType() string

+ func (s *Sub[T]) OutputShape() []int

+ func (s *Sub[T]) Parameters() []*graph.Parameter[T]

+ type Unsqueeze struct

+ func NewUnsqueeze[T tensor.Numeric](engine compute.Engine[T], axes []int) *Unsqueeze[T]

+ func (u *Unsqueeze[T]) Attributes() map[string]interface{}

+ func (u *Unsqueeze[T]) Backward(ctx context.Context, mode types.BackwardMode, ...) ([]*tensor.TensorNumeric[T], error)

+ func (u *Unsqueeze[T]) Forward(ctx context.Context, inputs ...*tensor.TensorNumeric[T]) (*tensor.TensorNumeric[T], error)

+ func (u *Unsqueeze[T]) OpType() string

+ func (u *Unsqueeze[T]) OutputShape() []int

+ func (u *Unsqueeze[T]) Parameters() []*graph.Parameter[T]

v0.1.0

Aug 4, 2025

Changes in this version

+ type Bias struct

+ func NewBiasWithFactories[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], size int, ...) (*Bias[T], error)

+ func NewBias[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], size int) (*Bias[T], error)

+ func (b *Bias[T]) Backward(outputGradient *tensor.Tensor[T]) ([]*tensor.Tensor[T], error)

+ func (b *Bias[T]) Forward(inputs ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (b *Bias[T]) OutputShape() []int

+ func (b *Bias[T]) Parameters() []*graph.Parameter[T]

+ func (b *Bias[T]) SetName(name string)

+ type Dense struct

+ func NewDense[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], ...) (*Dense[T], error)

+ func (d *Dense[T]) Backward(outputGradient *tensor.Tensor[T]) ([]*tensor.Tensor[T], error)

+ func (d *Dense[T]) Forward(inputs ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (d *Dense[T]) OutputShape() []int

+ func (d *Dense[T]) Parameters() []*graph.Parameter[T]

+ func (d *Dense[T]) SetName(name string)

+ type Linear struct

+ func NewLinearWithFactories[T tensor.Numeric](name string, engine compute.Engine[T], _ numeric.Arithmetic[T], ...) (*Linear[T], error)

+ func NewLinearWithHe[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], ...) (*Linear[T], error)

+ func NewLinearWithInitializer[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], ...) (*Linear[T], error)

+ func NewLinearWithUniform[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], ...) (*Linear[T], error)

+ func NewLinearWithXavier[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], ...) (*Linear[T], error)

+ func NewLinear[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], ...) (*Linear[T], error)

+ func (l *Linear[T]) Backward(outputGradient *tensor.Tensor[T]) ([]*tensor.Tensor[T], error)

+ func (l *Linear[T]) Forward(inputs ...*tensor.Tensor[T]) (*tensor.Tensor[T], error)

+ func (l *Linear[T]) OutputShape() []int

+ func (l *Linear[T]) Parameters() []*graph.Parameter[T]

+ func (l *Linear[T]) SetName(name string)

+ type PolynomialExpansion struct

+ func NewPolynomialExpansion[T tensor.Numeric](name string, engine compute.Engine[T], ops numeric.Arithmetic[T], ...) (*PolynomialExpansion[T], error)

+ func (p *PolynomialExpansion[T]) Backward(outputGradient *tensor.Tensor[T]) []*tensor.Tensor[T]

+ func (p *PolynomialExpansion[T]) Forward(inputs ...*tensor.Tensor[T]) *tensor.Tensor[T]

+ func (p *PolynomialExpansion[T]) GetDegree() int

+ func (p *PolynomialExpansion[T]) GetInputSize() int

+ func (p *PolynomialExpansion[T]) GetOutputSize() int

+ func (p *PolynomialExpansion[T]) GetTermIndices() [][]int

+ func (p *PolynomialExpansion[T]) HasBias() bool

+ func (p *PolynomialExpansion[T]) OutputShape() []int

+ func (p *PolynomialExpansion[T]) Parameters() []*tensor.Tensor[T]

+ func (p *PolynomialExpansion[T]) SetName(_ string)