Module Make.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
val calc_fans : int array -> float * float
val run : typ -> int array -> Optimise.Algodiff.t -> Optimise.Algodiff.t
val to_string : typ -> string
val to_name : unit -> string