Module Owl_neural_neuron.LSTM

module LSTM: sig .. end

type neuron_typ = {

	`mutable wxi : Owl_neural_neuron.t;`
	`mutable whi : Owl_neural_neuron.t;`
	`mutable wxc : Owl_neural_neuron.t;`
	`mutable whc : Owl_neural_neuron.t;`
	`mutable wxf : Owl_neural_neuron.t;`
	`mutable whf : Owl_neural_neuron.t;`
	`mutable wxo : Owl_neural_neuron.t;`
	`mutable who : Owl_neural_neuron.t;`
	`mutable bi : Owl_neural_neuron.t;`
	`mutable bc : Owl_neural_neuron.t;`
	`mutable bf : Owl_neural_neuron.t;`
	`mutable bo : Owl_neural_neuron.t;`
	`mutable c : Owl_neural_neuron.t;`
	`mutable h : Owl_neural_neuron.t;`
	`mutable init_typ : Owl_neural_neuron.Init.typ;`
	`mutable in_shape : int array;`
	`mutable out_shape : int array;`

}

val create : ?inputs:int -> int -> 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 -> Owl_neural_neuron.t array

val mkpri : neuron_typ -> Owl_algodiff.S.t array

val mkadj : neuron_typ -> Owl_algodiff.S.t array

val update : neuron_typ -> Owl_algodiff.S.t array -> unit

val run : Owl_algodiff.S.t -> neuron_typ -> Owl_algodiff.S.t

val to_string : neuron_typ -> string