module MaxPool2D:sig
..end
type
neuron_typ = {
|
mutable padding : |
|
mutable kernel : |
|
mutable stride : |
|
mutable in_shape : |
|
mutable out_shape : |
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