highway

Documentation

Index

• type Model
  • func New[T float.DType](in int, activation activation.Activation) *Model
  • func (m *Model) Forward(xs ...mat.Tensor) []mat.Tensor

Types

type Model

type Model struct {
	nn.Module
	WIn        *nn.Param
	BIn        *nn.Param
	WT         *nn.Param
	BT         *nn.Param
	Activation activation.Activation
}

Model contains the serializable parameters.

func New

func New[T float.DType](in int, activation activation.Activation) *Model

New returns a new model with parameters initialized to zeros.

func (*Model) Forward

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

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