Documentation ¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Model ¶
type Model struct { nn.BaseModel WInS nn.Param `spago:"type:weights"` WInR nn.Param `spago:"type:weights"` WRecS nn.Param `spago:"type:weights"` WRecR nn.Param `spago:"type:weights"` BS nn.Param `spago:"type:biases"` BR nn.Param `spago:"type:biases"` S nn.Param `spago:"type:weights"` R nn.Param `spago:"type:weights"` States []*State `spago:"scope:processor"` }
Model contains the serializable parameters.
func (*Model) Forward ¶
Forward performs the forward step for each input node and returns the result.
func (*Model) LastState ¶
LastState returns the last state of the recurrent network. It returns nil if there are no states.
func (*Model) SetInitialState ¶
SetInitialState sets the initial state of the recurrent network. It panics if one or more states are already present.
Click to show internal directories.
Click to hide internal directories.