Module Owl_neural_neuron.Conv2D

module Conv2D: sig .. end

type neuron_typ = {

	`mutable w : Owl_neural_neuron.t;`
	`mutable b : Owl_neural_neuron.t;`
	`mutable s : int array;`
	`mutable padding : Owl_algodiff.S.padding;`
	`mutable init_typ : Owl_neural_neuron.Init.typ;`
	`mutable in_shape : int array;`
	`mutable out_shape : int array;`

}

val create : Owl_algodiff.S.padding ->
       ?inputs:int array ->
       int array -> int array -> 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