package owl

  1. Overview
  2. Docs

Module Make.P

module Uniform : sig ... end
module Gaussian : sig ... end
module Exponential : sig ... end
module Gamma : sig ... end
module Beta : sig ... end
module Chi2 : sig ... end
module F : sig ... end
module Cauchy : sig ... end
module Lomax : sig ... end
module Weibull : sig ... end
module Laplace : sig ... end
module Gumbel1 : sig ... end
module Gumbel2 : sig ... end
module Logistic : sig ... end
module Lognormal : sig ... end
module Rayleigh : sig ... end
type dist = Owl_distribution.Make(A).dist =
  1. | Uniform of Uniform.t
  2. | Gaussian of Gaussian.t
  3. | Exponential of Exponential.t
  4. | Gamma of Gamma.t
  5. | Beta of Beta.t
  6. | Chi2 of Chi2.t
  7. | F of F.t
  8. | Cauchy of Cauchy.t
  9. | Lomax of Lomax.t
  10. | Weibull of Weibull.t
  11. | Laplace of Laplace.t
  12. | Gumbel1 of Gumbel1.t
  13. | Gumbel2 of Gumbel2.t
  14. | Logistic of Logistic.t
  15. | Lognormal of Lognormal.t
  16. | Rayleigh of Rayleigh.t
val prob : dist -> A.arr -> A.arr
val log_prob : dist -> A.arr -> A.arr
val cdf : dist -> A.arr -> A.arr
val logcdf : dist -> A.arr -> A.arr
val variable : ?name:string -> unit -> t
val assign_arr : t -> A.arr -> unit
val assign_elt : t -> A.elt -> unit
val to_arr : t -> A.arr
val to_elt : t -> A.elt
val of_arr : ?name:string -> A.arr -> t
val of_elt : ?name:string -> A.elt -> t
val eval : t -> unit
val pp_lazy : Format.formatter -> t -> unit
val to_trace : t list -> string
val to_dot : t list -> string
val copy : t array -> t array
val is_var : t -> bool
val is_const : t -> bool
val refnum : t -> int
val map : ?name:string -> (t array -> t) -> t array -> t
val tile : t -> int array -> t
val repeat : ?axis:int -> t -> int -> t
val concatenate : ?axis:int -> t array -> t
val abs : t -> t
val neg : t -> t
val conj : t -> t
val reci : t -> t
val signum : t -> t
val sqr : t -> t
val sqrt : t -> t
val cbrt : t -> t
val exp : t -> t
val exp2 : t -> t
val exp10 : t -> t
val expm1 : t -> t
val log : t -> t
val log2 : t -> t
val log10 : t -> t
val log1p : t -> t
val sin : t -> t
val cos : t -> t
val tan : t -> t
val asin : t -> t
val acos : t -> t
val atan : t -> t
val sinh : t -> t
val cosh : t -> t
val tanh : t -> t
val asinh : t -> t
val acosh : t -> t
val atanh : t -> t
val floor : t -> t
val ceil : t -> t
val round : t -> t
val trunc : t -> t
val fix : t -> t
val erf : t -> t
val erfc : t -> t
val relu : t -> t
val softplus : t -> t
val softsign : t -> t
val softmax : ?axis:int -> t -> t
val sigmoid : t -> t
val sum : ?axis:int -> t -> t
val prod : ?axis:int -> t -> t
val min : ?axis:int -> t -> t
val max : ?axis:int -> t -> t
val mean : ?axis:int -> t -> t
val var : ?axis:int -> t -> t
val std : ?axis:int -> t -> t
val l1norm : ?axis:int -> t -> t
val l2norm : ?axis:int -> t -> t
val cumsum : ?axis:int -> t -> t
val cumprod : ?axis:int -> t -> t
val cummin : ?axis:int -> t -> t
val cummax : ?axis:int -> t -> t
val sum' : t -> t
val prod' : t -> t
val min' : t -> t
val max' : t -> t
val mean' : t -> t
val var' : t -> t
val std' : t -> t
val l1norm' : t -> t
val l2norm' : t -> t
val l2norm_sqr' : t -> t
val add : t -> t -> t
val sub : t -> t -> t
val mul : t -> t -> t
val div : t -> t -> t
val pow : t -> t -> t
val dot : t -> t -> t
val atan2 : t -> t -> t
val hypot : t -> t -> t
val fmod : t -> t -> t
val min2 : t -> t -> t
val max2 : t -> t -> t
val add_scalar : t -> t -> t
val sub_scalar : t -> t -> t
val mul_scalar : t -> t -> t
val div_scalar : t -> t -> t
val pow_scalar : t -> t -> t
val atan2_scalar : t -> t -> t
val fmod_scalar : t -> t -> t
val scalar_add : t -> t -> t
val scalar_sub : t -> t -> t
val scalar_mul : t -> t -> t
val scalar_div : t -> t -> t
val scalar_pow : t -> t -> t
val scalar_atan2 : t -> t -> t
val scalar_fmod : t -> t -> t
val conv1d : ?padding:Owl_types.padding -> t -> t -> int array -> t
val conv2d : ?padding:Owl_types.padding -> t -> t -> int array -> t
val conv3d : ?padding:Owl_types.padding -> t -> t -> int array -> t
val max_pool1d : ?padding:Owl_types.padding -> t -> int array -> int array -> t
val max_pool2d : ?padding:Owl_types.padding -> t -> int array -> int array -> t
val max_pool3d : ?padding:Owl_types.padding -> t -> int array -> int array -> t
val avg_pool1d : ?padding:Owl_types.padding -> t -> int array -> int array -> t
val avg_pool2d : ?padding:Owl_types.padding -> t -> int array -> int array -> t
val avg_pool3d : ?padding:Owl_types.padding -> t -> int array -> int array -> t
val conv1d_backward_input : t -> t -> int array -> t -> t
val conv1d_backward_kernel : t -> t -> int array -> t -> t
val conv2d_backward_input : t -> t -> int array -> t -> t
val conv2d_backward_kernel : t -> t -> int array -> t -> t
val conv3d_backward_input : t -> t -> int array -> t -> t
val conv3d_backward_kernel : t -> t -> int array -> t -> t
val max_pool1d_backward : Owl_types.padding -> t -> int array -> int array -> t -> t
val max_pool2d_backward : Owl_types.padding -> t -> int array -> int array -> t -> t
val avg_pool1d_backward : Owl_types.padding -> t -> int array -> int array -> t -> t
val avg_pool2d_backward : Owl_types.padding -> t -> int array -> int array -> t -> t
val elt_equal : t -> t -> t
val elt_not_equal : t -> t -> t
val elt_less : t -> t -> t
val elt_greater : t -> t -> t
val elt_less_equal : t -> t -> t
val elt_greater_equal : t -> t -> t
val elt_equal_scalar : t -> t -> t
val elt_not_equal_scalar : t -> t -> t
val elt_less_scalar : t -> t -> t
val elt_greater_scalar : t -> t -> t
val elt_less_equal_scalar : t -> t -> t
val elt_greater_equal_scalar : t -> t -> t
val invalidate : t -> unit
val id : t -> int
val name : t -> string
val get_by_id : t -> int -> t
val get_by_name : t -> string -> t array
val uniform : a:t -> b:t -> t
val gaussian : mu:t -> sigma:t -> t
val sample : t -> int array -> t
val infer : 'a -> unit