package owl

  1. Overview
  2. Docs
package owl
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Owl_dense_ndarray_sSource

Types and constants
Sourcetype elt = float
Create N-dimensional array
Sourceval empty : int array -> arr
Sourceval create : int array -> elt -> arr
Sourceval init : int array -> (int -> elt) -> arr
Sourceval init_nd : int array -> (int array -> elt) -> arr
Sourceval zeros : int array -> arr
Sourceval ones : int array -> arr
Sourceval uniform : ?a:elt -> ?b:elt -> int array -> arr
Sourceval gaussian : ?mu:elt -> ?sigma:elt -> int array -> arr
Sourceval sequential : ?a:elt -> ?step:elt -> int array -> arr
Sourceval linspace : elt -> elt -> int -> arr
Sourceval logspace : ?base:float -> elt -> elt -> int -> arr
Sourceval bernoulli : ?p:float -> int array -> arr
Obtain basic properties
Sourceval shape : arr -> int array
Sourceval num_dims : arr -> int
Sourceval nth_dim : arr -> int -> int
Sourceval numel : arr -> int
Sourceval nnz : arr -> int
Sourceval density : arr -> float
Sourceval size_in_bytes : arr -> int
Sourceval same_shape : arr -> arr -> bool
Sourceval same_data : arr -> arr -> bool
Sourceval strides : arr -> int array
Sourceval slice_size : arr -> int array
Sourceval ind : arr -> int -> int array
Sourceval i1d : arr -> int array -> int
Manipulate a N-dimensional array
Sourceval get : arr -> int array -> elt
Sourceval set : arr -> int array -> elt -> unit
Sourceval get_index : arr -> int array array -> elt array
Sourceval set_index : arr -> int array array -> elt array -> unit
Sourceval get_fancy : Owl_types.index list -> arr -> arr
Sourceval set_fancy : Owl_types.index list -> arr -> arr -> unit
Sourceval get_slice : int list list -> arr -> arr
Sourceval set_slice : int list list -> arr -> arr -> unit
Sourceval sub_left : arr -> int -> int -> arr
Sourceval sub_ndarray : int array -> arr -> arr array
Sourceval slice_left : arr -> int array -> arr
Sourceval reset : arr -> unit
Sourceval fill : arr -> elt -> unit
Sourceval copy : arr -> arr
Sourceval resize : ?head:bool -> arr -> int array -> arr
Sourceval reshape : arr -> int array -> arr
Sourceval flatten : arr -> arr
Sourceval reverse : arr -> arr
Sourceval flip : ?axis:int -> arr -> arr
Sourceval rotate : arr -> int -> arr
Sourceval transpose : ?axis:int array -> arr -> arr
Sourceval swap : int -> int -> arr -> arr
Sourceval tile : arr -> int array -> arr
Sourceval repeat : arr -> int array -> arr
Sourceval concat_vertical : arr -> arr -> arr
Sourceval concat_horizontal : arr -> arr -> arr
Sourceval concat_vh : arr array array -> arr
Sourceval concatenate : ?axis:int -> arr array -> arr
Sourceval split : ?axis:int -> int array -> arr -> arr array
Sourceval split_vh : (int * int) array array -> arr -> arr array array
Sourceval squeeze : ?axis:int array -> arr -> arr
Sourceval expand : ?hi:bool -> arr -> int -> arr
Sourceval pad : ?v:elt -> int list list -> arr -> arr
Sourceval dropout : ?rate:float -> arr -> arr
Sourceval top : arr -> int -> int array array
Sourceval bottom : arr -> int -> int array array
Sourceval sort : arr -> arr
Sourceval draw : ?axis:int -> arr -> int -> arr * int array
Sourceval mmap : Unix.file_descr -> ?pos:int64 -> bool -> int array -> arr
Iterate array elements
Sourceval iteri : (int -> elt -> unit) -> arr -> unit
Sourceval iter : (elt -> unit) -> arr -> unit
Sourceval mapi : (int -> elt -> elt) -> arr -> arr
Sourceval map : (elt -> elt) -> arr -> arr
Sourceval foldi : ?axis:int -> (int -> elt -> elt -> elt) -> elt -> arr -> arr
Sourceval fold : ?axis:int -> (elt -> elt -> elt) -> elt -> arr -> arr
Sourceval scani : ?axis:int -> (int -> elt -> elt -> elt) -> arr -> arr
Sourceval scan : ?axis:int -> (elt -> elt -> elt) -> arr -> arr
Sourceval filteri : (int -> elt -> bool) -> arr -> int array
Sourceval filter : (elt -> bool) -> arr -> int array
Sourceval iter2i : (int -> elt -> elt -> unit) -> arr -> arr -> unit
Sourceval iter2 : (elt -> elt -> unit) -> arr -> arr -> unit
Sourceval map2i : (int -> elt -> elt -> elt) -> arr -> arr -> arr
Sourceval map2 : (elt -> elt -> elt) -> arr -> arr -> arr
Sourceval iteri_nd : (int array -> elt -> unit) -> arr -> unit
Sourceval mapi_nd : (int array -> elt -> elt) -> arr -> arr
Sourceval foldi_nd : ?axis:int -> (int array -> elt -> elt -> elt) -> elt -> arr -> arr
Sourceval scani_nd : ?axis:int -> (int array -> elt -> elt -> elt) -> arr -> arr
Sourceval filteri_nd : (int array -> elt -> bool) -> arr -> int array array
Sourceval iter2i_nd : (int array -> elt -> elt -> unit) -> arr -> arr -> unit
Sourceval map2i_nd : (int array -> elt -> elt -> elt) -> arr -> arr -> arr
Sourceval iteri_slice : ?axis:int -> (int -> arr -> unit) -> arr -> unit
Sourceval iter_slice : ?axis:int -> (arr -> unit) -> arr -> unit
Sourceval mapi_slice : ?axis:int -> (int -> arr -> 'c) -> arr -> 'c array
Sourceval map_slice : ?axis:int -> (arr -> 'c) -> arr -> 'c array
Sourceval filteri_slice : ?axis:int -> (int -> arr -> bool) -> arr -> arr array
Sourceval filter_slice : ?axis:int -> (arr -> bool) -> arr -> arr array
Sourceval foldi_slice : ?axis:int -> (int -> 'c -> arr -> 'c) -> 'c -> arr -> 'c
Sourceval fold_slice : ?axis:int -> ('c -> arr -> 'c) -> 'c -> arr -> 'c
Examine array elements or compare two arrays
Sourceval exists : (elt -> bool) -> arr -> bool
Sourceval not_exists : (elt -> bool) -> arr -> bool
Sourceval for_all : (elt -> bool) -> arr -> bool
Sourceval is_zero : arr -> bool
Sourceval is_positive : arr -> bool
Sourceval is_negative : arr -> bool
Sourceval is_nonpositive : arr -> bool
Sourceval is_nonnegative : arr -> bool
Sourceval is_normal : arr -> bool
Sourceval not_nan : arr -> bool
Sourceval not_inf : arr -> bool
Sourceval equal : arr -> arr -> bool
Sourceval not_equal : arr -> arr -> bool
Sourceval greater : arr -> arr -> bool
Sourceval less : arr -> arr -> bool
Sourceval greater_equal : arr -> arr -> bool
Sourceval less_equal : arr -> arr -> bool
Sourceval elt_equal : arr -> arr -> arr
Sourceval elt_not_equal : arr -> arr -> arr
Sourceval elt_less : arr -> arr -> arr
Sourceval elt_greater : arr -> arr -> arr
Sourceval elt_less_equal : arr -> arr -> arr
Sourceval elt_greater_equal : arr -> arr -> arr
Sourceval equal_scalar : arr -> elt -> bool
Sourceval not_equal_scalar : arr -> elt -> bool
Sourceval less_scalar : arr -> elt -> bool
Sourceval greater_scalar : arr -> elt -> bool
Sourceval less_equal_scalar : arr -> elt -> bool
Sourceval greater_equal_scalar : arr -> elt -> bool
Sourceval elt_equal_scalar : arr -> elt -> arr
Sourceval elt_not_equal_scalar : arr -> elt -> arr
Sourceval elt_less_scalar : arr -> elt -> arr
Sourceval elt_greater_scalar : arr -> elt -> arr
Sourceval elt_less_equal_scalar : arr -> elt -> arr
Sourceval elt_greater_equal_scalar : arr -> elt -> arr
Sourceval approx_equal : ?eps:float -> arr -> arr -> bool
Sourceval approx_equal_scalar : ?eps:float -> arr -> elt -> bool
Sourceval approx_elt_equal : ?eps:float -> arr -> arr -> arr
Sourceval approx_elt_equal_scalar : ?eps:float -> arr -> elt -> arr
Input/Output functions
Sourceval of_array : elt array -> int array -> arr
Sourceval to_array : arr -> elt array
Sourceval print : ?max_row:int -> ?max_col:int -> ?header:bool -> ?fmt:(elt -> string) -> arr -> unit
Sourceval save : arr -> string -> unit
Sourceval load : string -> arr
Unary mathematical operations
Sourceval sum : ?axis:int -> arr -> arr
Sourceval sum' : arr -> elt
Sourceval prod : ?axis:int -> arr -> arr
Sourceval prod' : arr -> elt
Sourceval mean : ?axis:int -> arr -> arr
Sourceval mean' : arr -> elt
Sourceval var : ?axis:int -> arr -> arr
Sourceval var' : arr -> elt
Sourceval std : ?axis:int -> arr -> arr
Sourceval std' : arr -> elt
Sourceval min : ?axis:int -> arr -> arr
Sourceval min' : arr -> elt
Sourceval max : ?axis:int -> arr -> arr
Sourceval max' : arr -> elt
Sourceval minmax : ?axis:int -> arr -> arr * arr
Sourceval minmax' : arr -> elt * elt
Sourceval min_i : arr -> elt * int array
Sourceval max_i : arr -> elt * int array
Sourceval minmax_i : arr -> (elt * int array) * (elt * int array)
Sourceval abs : arr -> arr
Sourceval abs2 : arr -> arr
Sourceval conj : arr -> arr
Sourceval neg : arr -> arr
Sourceval reci : arr -> arr
Sourceval reci_tol : ?tol:elt -> arr -> arr
Sourceval signum : arr -> arr
Sourceval sqr : arr -> arr
Sourceval sqrt : arr -> arr
Sourceval cbrt : arr -> arr
Sourceval exp : arr -> arr
Sourceval exp2 : arr -> arr
Sourceval exp10 : arr -> arr
Sourceval expm1 : arr -> arr
Sourceval log : arr -> arr
Sourceval log10 : arr -> arr
Sourceval log2 : arr -> arr
Sourceval log1p : arr -> arr
Sourceval sin : arr -> arr
Sourceval cos : arr -> arr
Sourceval tan : arr -> arr
Sourceval asin : arr -> arr
Sourceval acos : arr -> arr
Sourceval atan : arr -> arr
Sourceval sinh : arr -> arr
Sourceval cosh : arr -> arr
Sourceval tanh : arr -> arr
Sourceval asinh : arr -> arr
Sourceval acosh : arr -> arr
Sourceval atanh : arr -> arr
Sourceval floor : arr -> arr
Sourceval ceil : arr -> arr
Sourceval round : arr -> arr
Sourceval trunc : arr -> arr
Sourceval fix : arr -> arr
Sourceval modf : arr -> arr * arr
Sourceval erf : arr -> arr
Sourceval erfc : arr -> arr
Sourceval logistic : arr -> arr
Sourceval relu : arr -> arr
Sourceval elu : ?alpha:elt -> arr -> arr
Sourceval leaky_relu : ?alpha:elt -> arr -> arr
Sourceval softplus : arr -> arr
Sourceval softsign : arr -> arr
Sourceval softmax : ?axis:int -> arr -> arr
Sourceval sigmoid : arr -> arr
Sourceval log_sum_exp' : arr -> float
Sourceval l1norm : ?axis:int -> arr -> arr
Sourceval l1norm' : arr -> elt
Sourceval l2norm : ?axis:int -> arr -> arr
Sourceval l2norm' : arr -> elt
Sourceval l2norm_sqr : ?axis:int -> arr -> arr
Sourceval l2norm_sqr' : arr -> elt
Sourceval vecnorm : ?axis:int -> ?p:float -> arr -> arr
Sourceval vecnorm' : ?p:float -> arr -> elt
Sourceval cumsum : ?axis:int -> arr -> arr
Sourceval cumprod : ?axis:int -> arr -> arr
Sourceval cummin : ?axis:int -> arr -> arr
Sourceval cummax : ?axis:int -> arr -> arr
Sourceval diff : ?axis:int -> ?n:int -> arr -> arr
Sourceval lgamma : arr -> arr
Binary mathematical operations
Sourceval add : arr -> arr -> arr
Sourceval sub : arr -> arr -> arr
Sourceval mul : arr -> arr -> arr
Sourceval div : arr -> arr -> arr
Sourceval add_scalar : arr -> elt -> arr
Sourceval sub_scalar : arr -> elt -> arr
Sourceval mul_scalar : arr -> elt -> arr
Sourceval div_scalar : arr -> elt -> arr
Sourceval scalar_add : elt -> arr -> arr
Sourceval scalar_sub : elt -> arr -> arr
Sourceval scalar_mul : elt -> arr -> arr
Sourceval scalar_div : elt -> arr -> arr
Sourceval pow : arr -> arr -> arr
Sourceval scalar_pow : elt -> arr -> arr
Sourceval pow_scalar : arr -> elt -> arr
Sourceval atan2 : arr -> arr -> arr
Sourceval scalar_atan2 : elt -> arr -> arr
Sourceval atan2_scalar : arr -> elt -> arr
Sourceval hypot : arr -> arr -> arr
Sourceval min2 : arr -> arr -> arr
Sourceval max2 : arr -> arr -> arr
Sourceval fmod : arr -> arr -> arr
Sourceval fmod_scalar : arr -> elt -> arr
Sourceval scalar_fmod : elt -> arr -> arr
Sourceval ssqr' : arr -> elt -> elt
Sourceval ssqr_diff' : arr -> arr -> elt
Sourceval cross_entropy' : arr -> arr -> float
Sourceval clip_by_value : ?amin:elt -> ?amax:elt -> arr -> arr
Sourceval clip_by_l2norm : elt -> arr -> arr
Sourceval fma : arr -> arr -> arr -> arr
Sourceval conv1d : ?padding:Owl_types.padding -> arr -> arr -> int array -> arr
Sourceval conv2d : ?padding:Owl_types.padding -> arr -> arr -> int array -> arr
Sourceval conv3d : ?padding:Owl_types.padding -> arr -> arr -> int array -> arr
Sourceval dilated_conv1d : ?padding:Owl_types.padding -> arr -> arr -> int array -> int array -> arr
Sourceval dilated_conv2d : ?padding:Owl_types.padding -> arr -> arr -> int array -> int array -> arr
Sourceval dilated_conv3d : ?padding:Owl_types.padding -> arr -> arr -> int array -> int array -> arr
Sourceval transpose_conv1d : ?padding:Owl_types.padding -> arr -> arr -> int array -> arr
Sourceval transpose_conv2d : ?padding:Owl_types.padding -> arr -> arr -> int array -> arr
Sourceval transpose_conv3d : ?padding:Owl_types.padding -> arr -> arr -> int array -> arr
Sourceval max_pool1d : ?padding:Owl_types.padding -> arr -> int array -> int array -> arr
Sourceval max_pool2d : ?padding:Owl_types.padding -> arr -> int array -> int array -> arr
Sourceval max_pool3d : ?padding:Owl_types.padding -> arr -> int array -> int array -> arr
Sourceval avg_pool1d : ?padding:Owl_types.padding -> arr -> int array -> int array -> arr
Sourceval avg_pool2d : ?padding:Owl_types.padding -> arr -> int array -> int array -> arr
Sourceval avg_pool3d : ?padding:Owl_types.padding -> arr -> int array -> int array -> arr
Sourceval max_pool2d_argmax : ?padding:Owl_types.padding -> arr -> int array -> int array -> arr * (int64, Bigarray.int64_elt, Bigarray.c_layout) Bigarray.Genarray.t
Sourceval upsampling2d : arr -> int array -> arr
Sourceval conv1d_backward_input : arr -> arr -> int array -> arr -> arr
Sourceval conv1d_backward_kernel : arr -> arr -> int array -> arr -> arr
Sourceval conv2d_backward_input : arr -> arr -> int array -> arr -> arr
Sourceval conv2d_backward_kernel : arr -> arr -> int array -> arr -> arr
Sourceval conv3d_backward_input : arr -> arr -> int array -> arr -> arr
Sourceval conv3d_backward_kernel : arr -> arr -> int array -> arr -> arr
Sourceval dilated_conv1d_backward_input : arr -> arr -> int array -> int array -> arr -> arr
Sourceval dilated_conv1d_backward_kernel : arr -> arr -> int array -> int array -> arr -> arr
Sourceval dilated_conv2d_backward_input : arr -> arr -> int array -> int array -> arr -> arr
Sourceval dilated_conv2d_backward_kernel : arr -> arr -> int array -> int array -> arr -> arr
Sourceval dilated_conv3d_backward_input : arr -> arr -> int array -> int array -> arr -> arr
Sourceval dilated_conv3d_backward_kernel : arr -> arr -> int array -> int array -> arr -> arr
Sourceval transpose_conv1d_backward_input : arr -> arr -> int array -> arr -> arr
Sourceval transpose_conv1d_backward_kernel : arr -> arr -> int array -> arr -> arr
Sourceval transpose_conv2d_backward_input : arr -> arr -> int array -> arr -> arr
Sourceval transpose_conv2d_backward_kernel : arr -> arr -> int array -> arr -> arr
Sourceval transpose_conv3d_backward_input : arr -> arr -> int array -> arr -> arr
Sourceval transpose_conv3d_backward_kernel : arr -> arr -> int array -> arr -> arr
Sourceval max_pool1d_backward : Owl_types.padding -> arr -> int array -> int array -> arr -> arr
Sourceval max_pool2d_backward : Owl_types.padding -> arr -> int array -> int array -> arr -> arr
Sourceval max_pool3d_backward : Owl_types.padding -> arr -> int array -> int array -> arr -> arr
Sourceval avg_pool1d_backward : Owl_types.padding -> arr -> int array -> int array -> arr -> arr
Sourceval avg_pool2d_backward : Owl_types.padding -> arr -> int array -> int array -> arr -> arr
Sourceval avg_pool3d_backward : Owl_types.padding -> arr -> int array -> int array -> arr -> arr
Sourceval upsampling2d_backward : arr -> int array -> arr -> arr
Tensor Calculus
Sourceval contract1 : (int * int) array -> arr -> arr
Sourceval contract2 : (int * int) array -> arr -> arr -> arr
Experimental functions
Sourceval one_hot : int -> arr -> arr
Sourceval sum_slices : ?axis:int -> arr -> arr
Sourceval sum_reduce : ?axis:int array -> arr -> arr
Sourceval slide : ?axis:int -> ?ofs:int -> ?step:int -> window:int -> arr -> arr
Fucntions of in-place modification
Sourceval create_ : out:arr -> elt -> unit
Sourceval uniform_ : ?a:elt -> ?b:elt -> out:arr -> unit
Sourceval gaussian_ : ?mu:elt -> ?sigma:elt -> out:arr -> unit
Sourceval sequential_ : ?a:elt -> ?step:elt -> out:arr -> unit
Sourceval bernoulli_ : ?p:float -> out:arr -> unit
Sourceval zeros_ : out:arr -> unit
Sourceval ones_ : out:arr -> unit
Sourceval sort_ : arr -> unit
Sourceval one_hot_ : out:arr -> int -> arr -> unit
Sourceval get_fancy_ : out:arr -> Owl_types.index list -> arr -> unit
Sourceval set_fancy_ : out:arr -> Owl_types.index list -> arr -> arr -> unit
Sourceval get_slice_ : out:arr -> int list list -> arr -> unit
Sourceval set_slice_ : out:arr -> int list list -> arr -> arr -> unit
Sourceval copy_ : out:arr -> arr -> unit
Sourceval reshape_ : out:arr -> arr -> unit
Sourceval reverse_ : out:arr -> arr -> unit
Sourceval transpose_ : out:arr -> ?axis:int array -> arr -> unit
Sourceval repeat_ : out:arr -> arr -> int array -> unit
Sourceval tile_ : out:arr -> arr -> int array -> unit
Sourceval pad_ : out:arr -> ?v:elt -> int list list -> arr -> unit
Sourceval sum_ : out:arr -> axis:int -> arr -> unit
Sourceval min_ : out:arr -> axis:int -> arr -> unit
Sourceval max_ : out:arr -> axis:int -> arr -> unit
Sourceval add_ : ?out:arr -> arr -> arr -> unit
Sourceval sub_ : ?out:arr -> arr -> arr -> unit
Sourceval mul_ : ?out:arr -> arr -> arr -> unit
Sourceval div_ : ?out:arr -> arr -> arr -> unit
Sourceval pow_ : ?out:arr -> arr -> arr -> unit
Sourceval atan2_ : ?out:arr -> arr -> arr -> unit
Sourceval hypot_ : ?out:arr -> arr -> arr -> unit
Sourceval fmod_ : ?out:arr -> arr -> arr -> unit
Sourceval min2_ : ?out:arr -> arr -> arr -> unit
Sourceval max2_ : ?out:arr -> arr -> arr -> unit
Sourceval add_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval sub_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval mul_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval div_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval pow_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval atan2_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval fmod_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval scalar_add_ : ?out:arr -> elt -> arr -> unit
Sourceval scalar_sub_ : ?out:arr -> elt -> arr -> unit
Sourceval scalar_mul_ : ?out:arr -> elt -> arr -> unit
Sourceval scalar_div_ : ?out:arr -> elt -> arr -> unit
Sourceval scalar_pow_ : ?out:arr -> elt -> arr -> unit
Sourceval scalar_atan2_ : ?out:arr -> elt -> arr -> unit
Sourceval scalar_fmod_ : ?out:arr -> elt -> arr -> unit
Sourceval fma_ : ?out:arr -> arr -> arr -> arr -> unit
Sourceval clip_by_value_ : ?out:arr -> ?amin:elt -> ?amax:elt -> arr -> unit
Sourceval clip_by_l2norm_ : ?out:arr -> elt -> arr -> unit
Sourceval dot_ : ?transa:bool -> ?transb:bool -> ?alpha:elt -> ?beta:elt -> c:arr -> arr -> arr -> unit
Sourceval conj_ : ?out:arr -> arr -> unit
Sourceval abs_ : ?out:arr -> arr -> unit
Sourceval neg_ : ?out:arr -> arr -> unit
Sourceval reci_ : ?out:arr -> arr -> unit
Sourceval signum_ : ?out:arr -> arr -> unit
Sourceval sqr_ : ?out:arr -> arr -> unit
Sourceval sqrt_ : ?out:arr -> arr -> unit
Sourceval cbrt_ : ?out:arr -> arr -> unit
Sourceval exp_ : ?out:arr -> arr -> unit
Sourceval exp2_ : ?out:arr -> arr -> unit
Sourceval exp10_ : ?out:arr -> arr -> unit
Sourceval expm1_ : ?out:arr -> arr -> unit
Sourceval log_ : ?out:arr -> arr -> unit
Sourceval log2_ : ?out:arr -> arr -> unit
Sourceval log10_ : ?out:arr -> arr -> unit
Sourceval log1p_ : ?out:arr -> arr -> unit
Sourceval sin_ : ?out:arr -> arr -> unit
Sourceval cos_ : ?out:arr -> arr -> unit
Sourceval tan_ : ?out:arr -> arr -> unit
Sourceval asin_ : ?out:arr -> arr -> unit
Sourceval acos_ : ?out:arr -> arr -> unit
Sourceval atan_ : ?out:arr -> arr -> unit
Sourceval sinh_ : ?out:arr -> arr -> unit
Sourceval cosh_ : ?out:arr -> arr -> unit
Sourceval tanh_ : ?out:arr -> arr -> unit
Sourceval asinh_ : ?out:arr -> arr -> unit
Sourceval acosh_ : ?out:arr -> arr -> unit
Sourceval atanh_ : ?out:arr -> arr -> unit
Sourceval floor_ : ?out:arr -> arr -> unit
Sourceval ceil_ : ?out:arr -> arr -> unit
Sourceval round_ : ?out:arr -> arr -> unit
Sourceval trunc_ : ?out:arr -> arr -> unit
Sourceval fix_ : ?out:arr -> arr -> unit
Sourceval erf_ : ?out:arr -> arr -> unit
Sourceval erfc_ : ?out:arr -> arr -> unit
Sourceval relu_ : ?out:arr -> arr -> unit
Sourceval softplus_ : ?out:arr -> arr -> unit
Sourceval softsign_ : ?out:arr -> arr -> unit
Sourceval sigmoid_ : ?out:arr -> arr -> unit
Sourceval softmax_ : ?out:arr -> ?axis:int -> arr -> unit
Sourceval cumsum_ : ?out:arr -> ?axis:int -> arr -> unit
Sourceval cumprod_ : ?out:arr -> ?axis:int -> arr -> unit
Sourceval cummin_ : ?out:arr -> ?axis:int -> arr -> unit
Sourceval cummax_ : ?out:arr -> ?axis:int -> arr -> unit
Sourceval dropout_ : ?out:arr -> ?rate:float -> arr -> unit
Sourceval elt_equal_ : ?out:arr -> arr -> arr -> unit
Sourceval elt_not_equal_ : ?out:arr -> arr -> arr -> unit
Sourceval elt_less_ : ?out:arr -> arr -> arr -> unit
Sourceval elt_greater_ : ?out:arr -> arr -> arr -> unit
Sourceval elt_less_equal_ : ?out:arr -> arr -> arr -> unit
Sourceval elt_greater_equal_ : ?out:arr -> arr -> arr -> unit
Sourceval elt_equal_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval elt_not_equal_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval elt_less_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval elt_greater_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval elt_less_equal_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval elt_greater_equal_scalar_ : ?out:arr -> arr -> elt -> unit
Sourceval conv1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
Sourceval conv2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
Sourceval conv3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
Sourceval dilated_conv1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> int array -> unit
Sourceval dilated_conv2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> int array -> unit
Sourceval dilated_conv3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> int array -> unit
Sourceval transpose_conv1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
Sourceval transpose_conv2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
Sourceval transpose_conv3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> arr -> int array -> unit
Sourceval max_pool1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
Sourceval max_pool2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
Sourceval max_pool3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
Sourceval avg_pool1d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
Sourceval avg_pool2d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
Sourceval avg_pool3d_ : out:arr -> ?padding:Owl_types.padding -> arr -> int array -> int array -> unit
Sourceval upsampling2d_ : out:arr -> arr -> int array -> unit
Sourceval conv1d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval conv1d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval conv2d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval conv2d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval conv3d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval conv3d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval dilated_conv1d_backward_input_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
Sourceval dilated_conv1d_backward_kernel_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
Sourceval dilated_conv2d_backward_input_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
Sourceval dilated_conv2d_backward_kernel_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
Sourceval dilated_conv3d_backward_input_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
Sourceval dilated_conv3d_backward_kernel_ : out:arr -> arr -> arr -> int array -> int array -> arr -> unit
Sourceval transpose_conv1d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval transpose_conv1d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval transpose_conv2d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval transpose_conv2d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval transpose_conv3d_backward_input_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval transpose_conv3d_backward_kernel_ : out:arr -> arr -> arr -> int array -> arr -> unit
Sourceval max_pool1d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
Sourceval max_pool2d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
Sourceval max_pool3d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
Sourceval avg_pool1d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
Sourceval avg_pool2d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
Sourceval avg_pool3d_backward_ : out:arr -> Owl_types.padding -> arr -> int array -> int array -> arr -> unit
Sourceval upsampling2d_backward_ : out:arr -> arr -> int array -> arr -> unit
Sourceval fused_adagrad_ : ?out:arr -> rate:float -> eps:float -> arr -> unit
Matrix functions
Sourceval row : arr -> int -> arr
Sourceval col : arr -> int -> arr
Sourceval rows : arr -> int array -> arr
Sourceval cols : arr -> int array -> arr
Sourceval copy_row_to : arr -> arr -> int -> unit
Sourceval copy_col_to : arr -> arr -> int -> unit
Sourceval row_num : arr -> int
Sourceval col_num : arr -> int
Sourceval dot : arr -> arr -> arr
Sourceval trace : arr -> elt
Sourceval diag : ?k:int -> arr -> arr
Sourceval to_rows : arr -> arr array
Sourceval of_rows : arr array -> arr
Sourceval to_cols : arr -> arr array
Sourceval of_cols : arr array -> arr
Sourceval to_arrays : arr -> elt array array
Sourceval of_arrays : elt array array -> arr
Sourceval draw_rows : ?replacement:bool -> arr -> int -> arr * int array
Sourceval draw_cols : ?replacement:bool -> arr -> int -> arr * int array
Sourceval draw_rows2 : ?replacement:bool -> arr -> arr -> int -> arr * arr * int array
Sourceval draw_cols2 : ?replacement:bool -> arr -> arr -> int -> arr * arr * int array
Helper functions
Sourceval float_to_elt : float -> elt
Sourceval elt_to_float : elt -> float
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct