Module Owl_neural_neuron.FullyConnected

module FullyConnected: sig .. end

type neuron_typ = {

	`mutable w : Owl_neural_neuron.t;`
	`mutable b : 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 ->
       Owl_neural_neuron.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 -> 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