Module Neuron.Init

type typ =
  1. | Uniform of float * float
  2. | Gaussian of float * float
  3. | Standard
  4. | Tanh
  5. | GlorotNormal
  6. | GlorotUniform
  7. | LecunNormal
  8. | HeNormal
  9. | Custom of int array -> Optimise.Algodiff.t
    (*

    Initialisation types

    *)
val calc_fans : int array -> float * float

Calculate fan-in and fan-out of weights.

val run : typ -> int array -> Optimise.Algodiff.t -> Optimise.Algodiff.t

Execute the computation in this neuron.

val to_string : 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.