Make.Recurrent
type neuron_typ = {
mutable whh : Optimise.Algodiff.t;
mutable wxh : Optimise.Algodiff.t;
mutable why : Optimise.Algodiff.t;
mutable bh : Optimise.Algodiff.t;
mutable by : Optimise.Algodiff.t;
mutable h : Optimise.Algodiff.t;
mutable act : Activation.typ;
mutable init_typ : Init.typ;
mutable in_shape : int array;
mutable out_shape : int array;
}
val create :
?time_steps:int ->
?inputs:int ->
int ->
int ->
Activation.typ ->
Init.typ ->
neuron_typ
val connect : int array -> neuron_typ -> unit
val init : neuron_typ -> unit
val reset : neuron_typ -> unit
val mktag : int -> neuron_typ -> unit
val mkpar : neuron_typ -> Optimise.Algodiff.t array
val mkpri : neuron_typ -> Optimise.Algodiff.t array
val mkadj : neuron_typ -> Optimise.Algodiff.t array
val update : neuron_typ -> Optimise.Algodiff.t array -> unit
val copy : neuron_typ -> neuron_typ
val run : Optimise.Algodiff.t -> neuron_typ -> Optimise.Algodiff.t
val to_string : neuron_typ -> string