sig
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 -> Owl_neural_neuron.AvgPool2D.neuron_typ
val connect : int array -> Owl_neural_neuron.AvgPool2D.neuron_typ -> unit
val run :
Owl_algodiff.S.t ->
Owl_neural_neuron.AvgPool2D.neuron_typ -> Owl_algodiff.S.t
val to_string : Owl_neural_neuron.AvgPool2D.neuron_typ -> string
end