module Init: sig .. end
sig
end
type typ =
|
Uniform of float * float
float * float
Gaussian of float * float
Standard
Tanh
Custom of (int -> int -> float)
(int -> int -> float)
val run : typ -> int -> int -> Owl_algodiff.S.t
typ -> int -> int -> Owl_algodiff.S.t
val to_string : typ -> string
typ -> string