Versions in this module Expand all Collapse all v0 v0.3.0 Aug 25, 2025 Changes in this version + type HeInitializerOption func(*HeInitializerOptions[T]) + type HeInitializerOptions struct + type LinearGradientComputerOption func(*LinearGradientComputerOptions[T]) + type LinearGradientComputerOptions struct + type MatrixMultiplierOption func(*MatrixMultiplierOptions[T]) + type MatrixMultiplierOptions struct + type UniformInitializerOption func(*UniformInitializerOptions[T]) + func WithScale[T tensor.Numeric](scale float64) UniformInitializerOption[T] + type UniformInitializerOptions struct + Scale float64 + type XavierInitializerOption func(*XavierInitializerOptions[T]) + type XavierInitializerOptions struct v0.1.0 Aug 4, 2025 Changes in this version + type HeInitializer struct + func NewHeInitializer[T tensor.Numeric](ops numeric.Arithmetic[T]) *HeInitializer[T] + func (h *HeInitializer[T]) Initialize(inputSize, outputSize int) ([]T, error) + type LinearGradientComputer struct + func NewLinearGradientComputer[T tensor.Numeric](engine compute.Engine[T]) *LinearGradientComputer[T] + func (g *LinearGradientComputer[T]) ComputeBothGradients(ctx context.Context, input, weights, outputGradient *tensor.Tensor[T]) (*tensor.Tensor[T], *tensor.Tensor[T], error) + func (g *LinearGradientComputer[T]) ComputeInputGradient(ctx context.Context, weights, outputGradient *tensor.Tensor[T]) (*tensor.Tensor[T], error) + func (g *LinearGradientComputer[T]) ComputeWeightGradient(ctx context.Context, input, outputGradient *tensor.Tensor[T]) (*tensor.Tensor[T], error) + type MatrixMultiplier struct + func NewMatrixMultiplier[T tensor.Numeric](engine compute.Engine[T]) *MatrixMultiplier[T] + func (m *MatrixMultiplier[T]) Multiply(ctx context.Context, a, b *tensor.Tensor[T]) (*tensor.Tensor[T], error) + func (m *MatrixMultiplier[T]) MultiplyWithDestination(ctx context.Context, a, b, dst *tensor.Tensor[T]) (*tensor.Tensor[T], error) + func (m *MatrixMultiplier[T]) Transpose(ctx context.Context, a *tensor.Tensor[T]) (*tensor.Tensor[T], error) + func (m *MatrixMultiplier[T]) TransposeWithDestination(ctx context.Context, a, dst *tensor.Tensor[T]) (*tensor.Tensor[T], error) + type UniformInitializer struct + func NewUniformInitializer[T tensor.Numeric](ops numeric.Arithmetic[T], scale float64) *UniformInitializer[T] + func (u *UniformInitializer[T]) Initialize(inputSize, outputSize int) ([]T, error) + type WeightInitializer interface + Initialize func(inputSize, outputSize int) ([]T, error) + type XavierInitializer struct + func NewXavierInitializer[T tensor.Numeric](ops numeric.Arithmetic[T]) *XavierInitializer[T] + func (x *XavierInitializer[T]) Initialize(inputSize, outputSize int) ([]T, error)
Go to main page