Module Algodiff.Maths

val (+) : t -> t -> t
val (-) : t -> t -> t
val (*) : t -> t -> t
val (/) : t -> t -> t
val (*@) : t -> t -> t
val (**) : t -> t -> t
val add : t -> t -> t
val sub : t -> t -> t
val mul : t -> t -> t
val div : t -> t -> t
val kron : t -> t -> t
val dot : t -> t -> t
val pow : t -> t -> t
val atan2 : t -> t -> t
val min2 : t -> t -> t
val max2 : t -> t -> t
val cross_entropy : t -> t -> t
val inv : t -> t
val neg : t -> t
val abs : t -> t
val signum : t -> t
val floor : t -> t
val ceil : t -> t
val round : t -> t
val sqr : t -> t
val sqrt : t -> t
val log : t -> t
val log2 : t -> t
val log10 : t -> t
val exp : t -> t
val sin : t -> t
val cos : t -> t
val tan : t -> t
val sinh : t -> t
val cosh : t -> t
val tanh : t -> t
val asin : t -> t
val acos : t -> t
val atan : t -> t
val asinh : t -> t
val acosh : t -> t
val atanh : t -> t
val sum' : t -> t
val log_sum_exp' : t -> t
val log_sum_exp : ?axis:int -> ?keep_dims:bool -> t -> t
val sum : ?axis:int -> ?keep_dims:bool -> t -> t
val sum_reduce : ?axis:int array -> t -> t
val mean : t -> t
val transpose : ?axis:int array -> t -> t
val swap : int -> int -> t -> t
val l1norm' : t -> t
val l2norm' : t -> t
val l2norm_sqr' : t -> t
val sigmoid : t -> t
val relu : t -> t
val dawsn : t -> t
val softplus : t -> t
val softsign : t -> t
val softmax : ?axis:int -> t -> t
val reshape : t -> int array -> t
val flatten : t -> t
val get_item : t -> int -> int -> t
val get_row : t -> int -> t
val concat : axis:int -> t -> t -> t
val split : axis:int -> int array -> t -> t array
val of_arrays : t array array -> t
val to_arrays : t -> t array array
val concatenate : axis:int -> t array -> t
val stack : axis:int -> t array -> t
val get_slice : int list list -> t -> t
val set_slice : int list list -> t -> t -> t
val get_fancy : Owl_types.index list -> t -> t
val set_fancy : Owl_types.index list -> t -> t -> t
val diag : ?k:int -> t -> t
val diagm : ?k:int -> t -> t
val trace : t -> t
val triu : ?k:int -> t -> t
val tril : ?k:int -> t -> t