Owl_algodiff_primal_ops.D
include module type of struct include Owl_dense_ndarray.D end
include module type of struct include Owl_dense_ndarray_d end
include Owl_dense_ndarray_intf.Common with type elt := elt and type arr := arr
include Owl_base_dense_ndarray_intf.Common
with type elt := elt
with type arr := arr
val number : Owl_types_common.number
val empty : int array -> arr
val zeros : int array -> arr
val ones : int array -> arr
val bernoulli : ?p:float -> int array -> arr
val shape : arr -> int array
val numel : arr -> int
val strides : arr -> int array
Refer to :doc:`owl_dense_ndarray_generic`
val slice_size : arr -> int array
Refer to :doc:`owl_dense_ndarray_generic`
val reset : arr -> unit
val is_zero : arr -> bool
val is_positive : arr -> bool
val is_negative : arr -> bool
val is_nonpositive : arr -> bool
val is_nonnegative : arr -> bool
val is_normal : arr -> bool
val not_nan : arr -> bool
val not_inf : arr -> bool
val row_num : arr -> int
val col_num : arr -> int
val unit_basis : int -> int -> arr
unit_basis k n i
returns a unit basis vector with i
th element set to 1.
val num_dims : arr -> int
val nth_dim : arr -> int -> int
val nnz : arr -> int
val density : arr -> float
val size_in_bytes : arr -> int
val ind : arr -> int -> int array
val i1d : arr -> int array -> int
val get_fancy : Owl_types.index list -> arr -> arr
val set_fancy : Owl_types.index list -> arr -> arr -> unit
val top : arr -> int -> int array array
val bottom : arr -> int -> int array array
val argsort :
arr ->
(int64, Stdlib.Bigarray.int64_elt, Stdlib.Bigarray.c_layout)
Stdlib.Bigarray.Genarray.t
val mmap : Unix.file_descr -> ?pos:int64 -> bool -> int array -> arr
val save : out:string -> arr -> unit
val load : string -> arr
val save_npy : out:string -> arr -> unit
val load_npy : string -> arr
val bernoulli_ : ?p:float -> out:arr -> unit
val zeros_ : out:arr -> unit
val ones_ : out:arr -> unit
val sort_ : arr -> unit
val get_fancy_ : out:arr -> Owl_types.index list -> arr -> unit
val set_fancy_ : out:arr -> Owl_types.index list -> arr -> arr -> unit
include Owl_dense_ndarray_intf.Real with type elt := elt and type arr := arr
val log_sum_exp' : arr -> float
include Owl_dense_ndarray_intf.NN with type arr := arr
include Owl_base_dense_ndarray_intf.NN with type arr := arr
val conv1d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val conv2d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val conv3d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val dilated_conv1d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
int array ->
arr
val dilated_conv2d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
int array ->
arr
val dilated_conv3d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
int array ->
arr
val transpose_conv1d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val transpose_conv2d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val transpose_conv3d :
?padding:Owl_types_common.padding ->
arr ->
arr ->
int array ->
arr
val max_pool1d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val max_pool2d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val max_pool3d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val avg_pool1d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val avg_pool2d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val avg_pool3d :
?padding:Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr
val max_pool1d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val max_pool2d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val max_pool3d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val avg_pool1d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val avg_pool2d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val avg_pool3d_backward :
Owl_types_common.padding ->
arr ->
int array ->
int array ->
arr ->
arr
val max_pool2d_argmax :
?padding:Owl_types.padding ->
arr ->
int array ->
int array ->
arr
* (int64, Stdlib.Bigarray.int64_elt, Stdlib.Bigarray.c_layout)
Stdlib.Bigarray.Genarray.t
val conv1d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
arr ->
int array ->
unit
val conv2d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
arr ->
int array ->
unit
val conv3d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
arr ->
int array ->
unit
val dilated_conv1d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
arr ->
int array ->
int array ->
unit
val dilated_conv2d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
arr ->
int array ->
int array ->
unit
val dilated_conv3d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
arr ->
int array ->
int array ->
unit
val transpose_conv1d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
arr ->
int array ->
unit
val transpose_conv2d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
arr ->
int array ->
unit
val transpose_conv3d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
arr ->
int array ->
unit
val max_pool1d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
int array ->
int array ->
unit
val max_pool2d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
int array ->
int array ->
unit
val max_pool3d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
int array ->
int array ->
unit
val avg_pool1d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
int array ->
int array ->
unit
val avg_pool2d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
int array ->
int array ->
unit
val avg_pool3d_ :
out:arr ->
?padding:Owl_types.padding ->
arr ->
int array ->
int array ->
unit
val max_pool1d_backward_ :
out:arr ->
Owl_types.padding ->
arr ->
int array ->
int array ->
arr ->
unit
val max_pool2d_backward_ :
out:arr ->
Owl_types.padding ->
arr ->
int array ->
int array ->
arr ->
unit
val max_pool3d_backward_ :
out:arr ->
Owl_types.padding ->
arr ->
int array ->
int array ->
arr ->
unit
val avg_pool1d_backward_ :
out:arr ->
Owl_types.padding ->
arr ->
int array ->
int array ->
arr ->
unit
val avg_pool2d_backward_ :
out:arr ->
Owl_types.padding ->
arr ->
int array ->
int array ->
arr ->
unit
val avg_pool3d_backward_ :
out:arr ->
Owl_types.padding ->
arr ->
int array ->
int array ->
arr ->
unit
include Owl_dense_ndarray_intf.Distribution with type arr := arr
include module type of struct include Owl_dense_ndarray.Operator end
include sig ... end
val (+) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (-) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (*) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (/) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (+$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (-$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (*$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (/$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val ($+) :
'a ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val ($-) :
'a ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val ($*) :
'a ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val ($/) :
'a ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (=) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
bool
val (!=) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
bool
val (<>) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
bool
val (>) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
bool
val (<) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
bool
val (>=) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
bool
val (<=) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
bool
include sig ... end
val (=$) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
val (!=$) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
val (<>$) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
val (<$) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
val (>$) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
val (<=$) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
val (>=$) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
val (=.) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (!=.) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (<>.) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (<.) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (>.) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (<=.) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (>=.) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (=.$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (!=.$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (<>.$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (<.$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (>.$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (<=.$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (>=.$) :
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (=~) :
?eps:float ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
bool
val (=~$) : ?eps:float -> ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> bool
val (=~.) :
?eps:float ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (=~.$) :
?eps:float ->
('a, 'b) Owl_dense_ndarray_generic.t ->
'a ->
('a, 'b) Owl_dense_ndarray_generic.t
val (%) :
(float, 'a) Owl_dense_ndarray_generic.t ->
(float, 'a) Owl_dense_ndarray_generic.t ->
(float, 'a) Owl_dense_ndarray_generic.t
val (%$) :
(float, 'a) Owl_dense_ndarray_generic.t ->
float ->
(float, 'a) Owl_dense_ndarray_generic.t
val (**) :
(float, 'a) Owl_dense_ndarray_generic.t ->
(float, 'a) Owl_dense_ndarray_generic.t ->
(float, 'a) Owl_dense_ndarray_generic.t
val ($**) :
float ->
(float, 'a) Owl_dense_ndarray_generic.t ->
(float, 'a) Owl_dense_ndarray_generic.t
val (**$) :
(float, 'a) Owl_dense_ndarray_generic.t ->
float ->
(float, 'a) Owl_dense_ndarray_generic.t
val (+=) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
unit
val (-=) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
unit
val (*=) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
unit
val (/=) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
unit
val (+$=) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> unit
val (-$=) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> unit
val (*$=) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> unit
val (/$=) : ('a, 'b) Owl_dense_ndarray_generic.t -> 'a -> unit
val (@=) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (@||) :
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t ->
('a, 'b) Owl_dense_ndarray_generic.t
val (.!{;..}) :
('a, 'b) Owl_dense_ndarray_generic.t ->
Owl_types.index array ->
('a, 'b) Owl_dense_ndarray_generic.t
val (.!{;..}<-) :
('a, 'b) Owl_dense_ndarray_generic.t ->
Owl_types.index array ->
('a, 'b) Owl_dense_ndarray_generic.t ->
unit
val (.${}) :
('a, 'b) Owl_dense_ndarray_generic.t ->
int list ->
('a, 'b) Owl_dense_ndarray_generic.t
val (.${;..}) :
('a, 'b) Owl_dense_ndarray_generic.t ->
int list array ->
('a, 'b) Owl_dense_ndarray_generic.t
val (.${}<-) :
('a, 'b) Owl_dense_ndarray_generic.t ->
int list ->
('a, 'b) Owl_dense_ndarray_generic.t ->
unit
val (.${;..}<-) :
('a, 'b) Owl_dense_ndarray_generic.t ->
int list array ->
('a, 'b) Owl_dense_ndarray_generic.t ->
unit
include sig ... end
val (.%{}) : ('a, 'b) Owl_dense_ndarray_generic.t -> int -> 'a
val (.%{;..}) : ('a, 'b) Owl_dense_ndarray_generic.t -> int array -> 'a
val (.%{}<-) : ('a, 'b) Owl_dense_ndarray_generic.t -> int -> 'a -> unit
val (.%{;..}<-) :
('a, 'b) Owl_dense_ndarray_generic.t ->
int array ->
'a ->
unit
val mpow : Owl_linalg_d.mat -> float -> Owl_linalg_d.mat
module Scalar = Owl_maths
module Mat : sig ... end
module Linalg : sig ... end