activation

Documentation

Index

• type Model
  • func New(activation ag.OpName, params ...nn.Param) *Model
  • func (m *Model) Forward(xs ...ag.Node) []ag.Node

Types

type Model

type Model struct {
	nn.BaseModel
	Activation ag.OpName
	Params     []nn.Param
}

Model contains the activation operator and serializable parameters.

func New

func New(activation ag.OpName, params ...nn.Param) *Model

New returns a new model with parameters initialized to zeros. TODO: restrict operators to activation functions only; or create a dedicate builder for each activation.

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.