Module Neuron.AvgPool2D

type neuron_typ = {
  1. mutable padding : Owl_types.padding;
  2. mutable kernel : int array;
  3. mutable stride : int array;
  4. mutable in_shape : int array;
  5. mutable out_shape : int array;
}

Neuron type definition.

val create : Owl_types.padding -> int array -> int array -> neuron_typ

Create the neuron.

val connect : int array -> neuron_typ -> unit

Connect this neuron to others in a neural network.

val copy : neuron_typ -> neuron_typ

Make a deep copy of the neuron and its parameters.

Execute the computation in this neuron.

val to_string : neuron_typ -> string

Convert the neuron to its string representation. The string is often a summary of the parameters defined in the neuron.

val to_name : unit -> string

Return the name of the neuron.