Module Owl_neural_neuron.AvgPool2D

module AvgPool2D: sig .. end

type neuron_typ = {

	`mutable padding : Owl_algodiff.S.padding;`
	`mutable kernel : int array;`
	`mutable stride : int array;`
	`mutable in_shape : int array;`
	`mutable out_shape : int array;`

}

val create : Owl_algodiff.S.padding ->
       int array -> int array -> neuron_typ

val connect : int array -> neuron_typ -> unit

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

val to_string : neuron_typ -> string