Module Neuron.Max

type neuron_typ = {
  1. mutable in_shape : int array;
  2. mutable out_shape : int array;
}

Neuron type definition.

val create : unit -> neuron_typ

Create the neuron.

val connect : int array array -> neuron_typ -> unit

Connect this neuron to others in a neural network.

val copy : 'a -> neuron_typ

Make a deep copy of the neuron and its parameters.

val run : Optimise.Algodiff.t array -> 'a -> Optimise.Algodiff.t

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.