sig
  type ('a, 'b) t = ('a, 'b, Bigarray.c_layout) Bigarray.Genarray.t
  type ('a, 'b) kind = ('a, 'b) Bigarray.kind
  val empty :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val create :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    int array -> 'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val init :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    int array -> (int -> 'a) -> ('a, 'b) Owl_dense_ndarray_generic.t
  val init_nd :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    int array -> (int array -> 'a) -> ('a, 'b) Owl_dense_ndarray_generic.t
  val zeros :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val ones :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val uniform :
    ?scale:float ->
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val gaussian :
    ?sigma:float ->
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val sequential :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    ?a:'a -> ?step:'a -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val linspace :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    'a -> 'a -> int -> ('a, 'b) Owl_dense_ndarray_generic.t
  val logspace :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    ?base:float -> 'a -> 'a -> int -> ('a, 'b) Owl_dense_ndarray_generic.t
  val bernoulli :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    ?p:float ->
    ?seed:int -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val shape : ('a, 'b) Owl_dense_ndarray_generic.t -> int array
  val num_dims : ('a, 'b) Owl_dense_ndarray_generic.t -> int
  val nth_dim : ('a, 'b) Owl_dense_ndarray_generic.t -> int -> int
  val numel : ('a, 'b) Owl_dense_ndarray_generic.t -> int
  val nnz : ('a, 'b) Owl_dense_ndarray_generic.t -> int
  val density : ('a, 'b) Owl_dense_ndarray_generic.t -> float
  val size_in_bytes : ('a, 'b) Owl_dense_ndarray_generic.t -> int
  val same_shape :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val kind :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.kind
  val strides : ('a, 'b) Owl_dense_ndarray_generic.t -> int array
  val slice_size : ('a, 'b) Owl_dense_ndarray_generic.t -> int array
  val index_1d_nd : int -> int array -> int array
  val index_nd_1d : int array -> int array -> int
  val get : ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> 'a
  val set : ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> 'a -> unit
  val sub_left :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    int -> int -> ('a, 'b) Owl_dense_ndarray_generic.t
  val slice_left :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val slice :
    int list list ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val copy :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> unit
  val reset : ('a, 'b) Owl_dense_ndarray_generic.t -> unit
  val fill : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> unit
  val clone :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val resize :
    ?head:bool ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val reshape :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val flatten :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val reverse :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val transpose :
    ?axis:int array ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val swap :
    int ->
    int ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val tile :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val repeat :
    ?axis:int ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    int -> ('a, 'b) Owl_dense_ndarray_generic.t
  val concatenate :
    ?axis:int ->
    ('a, 'b) Owl_dense_ndarray_generic.t array ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val squeeze :
    ?axis:int array ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val expand :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    int -> ('a, 'b) Owl_dense_ndarray_generic.t
  val pad :
    ?v:'a ->
    int list list ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val dropout :
    ?rate:float ->
    ?seed:int ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val mmap :
    Unix.file_descr ->
    ?pos:int64 ->
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    bool -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val iteri :
    ?axis:int option array ->
    (int array -> 'a -> unit) -> ('a, 'b) Owl_dense_ndarray_generic.t -> unit
  val iter :
    ?axis:int option array ->
    ('a -> unit) -> ('a, 'b) Owl_dense_ndarray_generic.t -> unit
  val mapi :
    ?axis:int option array ->
    (int array -> 'a -> 'a) ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val map :
    ?axis:int option array ->
    ('a -> 'a) ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val map2i :
    ?axis:int option array ->
    (int array -> 'a -> 'a -> 'a) ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val map2 :
    ?axis:int option array ->
    ('a -> 'a -> 'a) ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val filteri :
    ?axis:int option array ->
    (int array -> 'a -> bool) ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> int array array
  val filter :
    ?axis:int option array ->
    ('a -> bool) -> ('a, 'b) Owl_dense_ndarray_generic.t -> int array array
  val foldi :
    ?axis:int option array ->
    (int array -> 'c -> 'a -> 'c) ->
    'c -> ('a, 'b) Owl_dense_ndarray_generic.t -> 'c
  val fold :
    ?axis:int option array ->
    ('c -> 'a -> 'c) -> 'c -> ('a, 'b) Owl_dense_ndarray_generic.t -> 'c
  val iteri_slice :
    int array ->
    (int array array -> ('a, 'b) Owl_dense_ndarray_generic.t -> unit) ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> unit
  val iter_slice :
    int array ->
    (('a, 'b) Owl_dense_ndarray_generic.t -> unit) ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> unit
  val iter2i :
    (int array -> 'a -> 'b -> unit) ->
    ('a, 'c) Owl_dense_ndarray_generic.t ->
    ('b, 'd) Owl_dense_ndarray_generic.t -> unit
  val iter2 :
    ('a -> 'b -> unit) ->
    ('a, 'c) Owl_dense_ndarray_generic.t ->
    ('b, 'd) Owl_dense_ndarray_generic.t -> unit
  val exists : ('a -> bool) -> ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val not_exists :
    ('a -> bool) -> ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val for_all : ('a -> bool) -> ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val is_zero : ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val is_positive : ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val is_negative : ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val is_nonpositive : ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val is_nonnegative : ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val is_normal : ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val not_nan : ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val not_inf : ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val equal :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val not_equal :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val greater :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val less :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val greater_equal :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val less_equal :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val elt_equal :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_not_equal :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_less :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_greater :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_less_equal :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_greater_equal :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val equal_scalar : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
  val not_equal_scalar : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
  val less_scalar : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
  val greater_scalar : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
  val less_equal_scalar : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
  val greater_equal_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
  val elt_equal_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_not_equal_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_less_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_greater_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_less_equal_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val elt_greater_equal_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val approx_equal :
    ?eps:float ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> bool
  val approx_equal_scalar :
    ?eps:float -> ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
  val approx_elt_equal :
    ?eps:float ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val approx_elt_equal_scalar :
    ?eps:float ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val of_array :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    'a array -> int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val to_array : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a array
  val print : ('a, 'b) Owl_dense_ndarray_generic.t -> unit
  val pp_dsnda : ('a, 'b) Owl_dense_ndarray_generic.t -> unit
  val save : ('a, 'b) Owl_dense_ndarray_generic.t -> string -> unit
  val load :
    ('a, 'b) Owl_dense_ndarray_generic.kind ->
    string -> ('a, 'b) Owl_dense_ndarray_generic.t
  val re_c2s :
    (Complex.t, Bigarray.complex32_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float32_elt) Owl_dense_ndarray_generic.t
  val re_z2d :
    (Complex.t, Bigarray.complex64_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float64_elt) Owl_dense_ndarray_generic.t
  val im_c2s :
    (Complex.t, Bigarray.complex32_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float32_elt) Owl_dense_ndarray_generic.t
  val im_z2d :
    (Complex.t, Bigarray.complex64_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float64_elt) Owl_dense_ndarray_generic.t
  val sum : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a
  val prod :
    ?axis:int option array -> ('a, 'b) Owl_dense_ndarray_generic.t -> 'a
  val min : (float, 'a) Owl_dense_ndarray_generic.t -> float
  val max : (float, 'a) Owl_dense_ndarray_generic.t -> float
  val minmax : (float, 'a) Owl_dense_ndarray_generic.t -> float * float
  val min_i : (float, 'a) Owl_dense_ndarray_generic.t -> float * int array
  val max_i : (float, 'a) Owl_dense_ndarray_generic.t -> float * int array
  val minmax_i :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float * int array) * (float * int array)
  val abs :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val abs_c2s :
    (Complex.t, Bigarray.complex32_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float32_elt) Owl_dense_ndarray_generic.t
  val abs_z2d :
    (Complex.t, Bigarray.complex64_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float64_elt) Owl_dense_ndarray_generic.t
  val abs2 :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val abs2_c2s :
    (Complex.t, Bigarray.complex32_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float32_elt) Owl_dense_ndarray_generic.t
  val abs2_z2d :
    (Complex.t, Bigarray.complex64_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float64_elt) Owl_dense_ndarray_generic.t
  val conj :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val neg :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val reci :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val signum :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val sqr :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val sqrt :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val cbrt :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val exp :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val exp2 :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val exp10 :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val expm1 :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val log :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val log10 :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val log2 :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val log1p :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val sin :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val cos :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val tan :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val asin :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val acos :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val atan :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val sinh :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val cosh :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val tanh :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val asinh :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val acosh :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val atanh :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val floor :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val ceil :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val round :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val trunc :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val modf :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t *
    ('a, 'b) Owl_dense_ndarray_generic.t
  val erf :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val erfc :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val logistic :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val relu :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val elu :
    ?alpha:float ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val leaky_relu :
    ?alpha:float ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val softplus :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val softsign :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val softmax :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val sigmoid :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val log_sum_exp : (float, 'a) Owl_dense_ndarray_generic.t -> float
  val l1norm : ('a, 'b) Owl_dense_ndarray_generic.t -> float
  val l2norm : ('a, 'b) Owl_dense_ndarray_generic.t -> float
  val l2norm_sqr : ('a, 'b) Owl_dense_ndarray_generic.t -> float
  val cumsum :
    ?axis:int ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val cumprod :
    ?axis:int ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val add :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val sub :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val mul :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val div :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val add_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val sub_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val mul_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val div_scalar :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    'a -> ('a, 'b) Owl_dense_ndarray_generic.t
  val scalar_add :
    'a ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val scalar_sub :
    'a ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val scalar_mul :
    'a ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val scalar_div :
    'a ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val pow :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val scalar_pow :
    float ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val pow_scalar :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    float -> (float, 'a) Owl_dense_ndarray_generic.t
  val atan2 :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val scalar_atan2 :
    float ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val atan2_scalar :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    float -> (float, 'a) Owl_dense_ndarray_generic.t
  val hypot :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val min2 :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val max2 :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val fmod :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val fmod_scalar :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    float -> (float, 'a) Owl_dense_ndarray_generic.t
  val scalar_fmod :
    float ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val ssqr : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> 'a
  val ssqr_diff :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t -> 'a
  val cross_entropy :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t -> float
  val clip_by_l2norm :
    float ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val cast_s2d :
    (float, Bigarray.float32_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float64_elt) Owl_dense_ndarray_generic.t
  val cast_d2s :
    (float, Bigarray.float64_elt) Owl_dense_ndarray_generic.t ->
    (float, Bigarray.float32_elt) Owl_dense_ndarray_generic.t
  val cast_c2z :
    (Complex.t, Bigarray.complex32_elt) Owl_dense_ndarray_generic.t ->
    (Complex.t, Bigarray.complex64_elt) Owl_dense_ndarray_generic.t
  val cast_z2c :
    (Complex.t, Bigarray.complex64_elt) Owl_dense_ndarray_generic.t ->
    (Complex.t, Bigarray.complex32_elt) Owl_dense_ndarray_generic.t
  val cast_s2c :
    (float, Bigarray.float32_elt) Owl_dense_ndarray_generic.t ->
    (Complex.t, Bigarray.complex32_elt) Owl_dense_ndarray_generic.t
  val cast_d2z :
    (float, Bigarray.float64_elt) Owl_dense_ndarray_generic.t ->
    (Complex.t, Bigarray.complex64_elt) Owl_dense_ndarray_generic.t
  val cast_s2z :
    (float, Bigarray.float32_elt) Owl_dense_ndarray_generic.t ->
    (Complex.t, Bigarray.complex64_elt) Owl_dense_ndarray_generic.t
  val cast_d2c :
    (float, Bigarray.float64_elt) Owl_dense_ndarray_generic.t ->
    (Complex.t, Bigarray.complex32_elt) Owl_dense_ndarray_generic.t
  type padding = SAME | VALID
  val conv1d :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array -> (float, 'a) Owl_dense_ndarray_generic.t
  val conv2d :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array -> (float, 'a) Owl_dense_ndarray_generic.t
  val conv3d :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array -> (float, 'a) Owl_dense_ndarray_generic.t
  val max_pool1d :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array -> int array -> (float, 'a) Owl_dense_ndarray_generic.t
  val max_pool2d :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array -> int array -> (float, 'a) Owl_dense_ndarray_generic.t
  val max_pool3d :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array -> int array -> (float, 'a) Owl_dense_ndarray_generic.t
  val avg_pool1d :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array -> int array -> (float, 'a) Owl_dense_ndarray_generic.t
  val avg_pool2d :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array -> int array -> (float, 'a) Owl_dense_ndarray_generic.t
  val avg_pool3d :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array -> int array -> (float, 'a) Owl_dense_ndarray_generic.t
  val max_pool2d_argmax :
    ?padding:Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array ->
    int array ->
    (float, 'a) Owl_dense_ndarray_generic.t *
    (int64, Bigarray.int64_elt) Owl_dense_ndarray_generic.t
  val conv2d_backward_input :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val conv2d_backward_kernel :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val conv3d_backward_input :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val conv3d_backward_kernel :
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val max_pool2d_backward :
    Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array ->
    int array ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val avg_pool2d_backward :
    Owl_dense_ndarray_generic.padding ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    int array ->
    int array ->
    (float, 'a) Owl_dense_ndarray_generic.t ->
    (float, 'a) Owl_dense_ndarray_generic.t
  val print_element : ('a, 'b) Owl_dense_ndarray_generic.kind -> 'a -> unit
  val print_index : int array -> unit
  val _check_transpose_axis : int array -> int -> unit
  val sum_slices :
    ?axis:int ->
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    ('a, 'b) Owl_dense_ndarray_generic.t
  val calc_conv2d_output_shape :
    Owl_dense_ndarray_generic.padding ->
    int -> int -> int -> int -> int -> int -> int * int
  val calc_conv3d_output_shape :
    Owl_dense_ndarray_generic.padding ->
    int ->
    int -> int -> int -> int -> int -> int -> int -> int -> int * int * int
  val slice_along_dim0 :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    int array -> ('a, 'b) Owl_dense_ndarray_generic.t
  val draw_along_dim0 :
    ('a, 'b) Owl_dense_ndarray_generic.t ->
    int -> ('a, 'b) Owl_dense_ndarray_generic.t * int array
end