batchnorm

package

v1.1.0 Latest Latest Go to latest Published: Oct 30, 2023 License: BSD-2-Clause Imports: 5 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Model struct {
	nn.Module
	W        *nn.Param
	B        *nn.Param
	Mean     *nn.Buffer
	StdDev   *nn.Buffer
	Momentum *nn.Buffer
}

Model contains the serializable parameters.

func New[T float.DType](size int) *Model

New returns a new model with the supplied size and default momentum

func NewWithMomentum[T float.DType](size int, momentum T) *Model

NewWithMomentum returns a new model with supplied size and momentum.

func (m *Model) Forward(xs ...mat.Tensor) []mat.Tensor

Forward performs the forward step for each input node and returns the result.

func (m *Model) ForwardT(xs ...mat.Tensor) []mat.Tensor

ForwardT performs the forward step for each input node and returns the result.