batchnorm

Documentation

Index

• type Model
  • func New(size int) *Model
  • func NewWithMomentum(size int, momentum mat.Float) *Model
  • func (m *Model) Forward(xs ...ag.Node) []ag.Node

Types

type Model

type Model struct {
	nn.BaseModel
	W        nn.Param `spago:"type:weights"`
	B        nn.Param `spago:"type:biases"`
	Mean     nn.Param `spago:"type:undefined"`
	StdDev   nn.Param `spago:"type:undefined"`
	Momentum nn.Param `spago:"type:undefined"`
}

Model contains the serializable parameters.

func New

func New(size int) *Model

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

func NewWithMomentum

func NewWithMomentum(size int, momentum mat.Float) *Model

NewWithMomentum returns a new model with supplied size and momentum.

func (*Model) Forward

func (m *Model) Forward(xs ...ag.Node) []ag.Node

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