Module Neuron.LSTM

type neuron_typ = Owl_neural_neuron.Make(Owl_optimise_generic.Make(Owl_algodiff_generic.Make(A))).LSTM.neuron_typ = {
  1. mutable wxi : Optimise.Algodiff.t;
  2. mutable whi : Optimise.Algodiff.t;
  3. mutable wxc : Optimise.Algodiff.t;
  4. mutable whc : Optimise.Algodiff.t;
  5. mutable wxf : Optimise.Algodiff.t;
  6. mutable whf : Optimise.Algodiff.t;
  7. mutable wxo : Optimise.Algodiff.t;
  8. mutable who : Optimise.Algodiff.t;
  9. mutable bi : Optimise.Algodiff.t;
  10. mutable bc : Optimise.Algodiff.t;
  11. mutable bf : Optimise.Algodiff.t;
  12. mutable bo : Optimise.Algodiff.t;
  13. mutable c : Optimise.Algodiff.t;
  14. mutable h : Optimise.Algodiff.t;
  15. mutable init_typ : Init.typ;
  16. mutable in_shape : int array;
  17. mutable out_shape : int array;
}
val create : ?time_steps:int -> ?inputs:int -> int -> 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 to_string : neuron_typ -> string
val to_name : unit -> string