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