Module `Algodiff.A`Source

Source

type arr =
  Owl_neural_generic.Make_Embedded(Owl_dense_ndarray.S).Neuron.Optimise.Algodiff.A.arr

Source

type elt =
  Owl_neural_generic.Make_Embedded(Owl_dense_ndarray.S).Neuron.Optimise.Algodiff.A.elt

Sourceval number : Owl_types_common.number

Sourceval empty : int array -> arr

Sourceval zeros : int array -> arr

Sourceval ones : int array -> arr

Sourceval create : int array -> elt -> arr

Sourceval sequential : ?a:elt -> ?step:elt -> int array -> arr

Sourceval uniform : ?a:elt -> ?b:elt -> int array -> arr

Sourceval gaussian : ?mu:elt -> ?sigma:elt -> int array -> arr

Sourceval bernoulli : ?p:elt -> int array -> arr

Sourceval init : int array -> (int -> elt) -> arr

Sourceval init_nd : int array -> (int array -> elt) -> arr

Sourceval shape : arr -> int array

Sourceval numel : arr -> int

Sourceval get : arr -> int array -> elt

Sourceval set : arr -> int array -> elt -> unit

Sourceval get_slice : int list list -> arr -> arr

Sourceval set_slice : int list list -> arr -> arr -> unit

Sourceval copy : arr -> arr

Sourceval copy_ : out:arr -> arr -> unit

Sourceval reset : arr -> unit

Sourceval reshape : arr -> int array -> arr

Sourceval reverse : arr -> arr

Sourceval tile : arr -> int array -> arr

Sourceval repeat : arr -> int array -> arr

Sourceval concatenate : ?axis:int -> arr array -> arr

Sourceval split : ?axis:int -> int array -> arr -> arr array

Sourceval draw : ?axis:int -> arr -> int -> arr * int array

Sourceval map : (elt -> elt) -> arr -> arr

Sourceval fold : ?axis:int -> (elt -> elt -> elt) -> elt -> arr -> arr

Sourceval scan : ?axis:int -> (elt -> elt -> elt) -> arr -> arr

Sourceval one_hot : int -> arr -> arr

Sourceval pad : ?v:elt -> int list list -> arr -> arr

Source

val print : 
  ?max_row:int ->
  ?max_col:int ->
  ?header:bool ->
  ?fmt:(elt -> string) ->
  arr ->
  unit

Sourceval abs : arr -> arr

Sourceval neg : arr -> arr

Sourceval floor : arr -> arr

Sourceval ceil : arr -> arr

Sourceval round : arr -> arr

Sourceval sqr : arr -> arr

Sourceval sqrt : arr -> arr

Sourceval log : arr -> arr

Sourceval log2 : arr -> arr

Sourceval log10 : arr -> arr

Sourceval exp : arr -> arr

Sourceval sin : arr -> arr

Sourceval cos : arr -> arr

Sourceval tan : arr -> arr

Sourceval sinh : arr -> arr

Sourceval cosh : arr -> arr

Sourceval tanh : arr -> arr

Sourceval asin : arr -> arr

Sourceval acos : arr -> arr

Sourceval atan : arr -> arr

Sourceval asinh : arr -> arr

Sourceval acosh : arr -> arr

Sourceval atanh : arr -> arr

Sourceval min : ?axis:int -> arr -> arr

Sourceval max : ?axis:int -> arr -> arr

Sourceval sum : ?axis:int -> arr -> arr

Sourceval sum_reduce : ?axis:int array -> arr -> arr

Sourceval signum : arr -> arr

Sourceval sigmoid : arr -> arr

Sourceval relu : arr -> arr

Sourceval min' : arr -> elt

Sourceval max' : arr -> elt

Sourceval sum' : arr -> elt

Sourceval l1norm' : arr -> elt

Sourceval l2norm' : arr -> elt

Sourceval l2norm_sqr' : arr -> elt

Sourceval clip_by_value : ?amin:elt -> ?amax:elt -> arr -> arr

Sourceval clip_by_l2norm : 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 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 fma : arr -> arr -> arr -> arr

Source

val conv1d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  arr ->
  int array ->
  arr

Source

val conv2d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  arr ->
  int array ->
  arr

Source

val conv3d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  arr ->
  int array ->
  arr

Source

val dilated_conv1d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  arr ->
  int array ->
  int array ->
  arr

Source

val dilated_conv2d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  arr ->
  int array ->
  int array ->
  arr

Source

val dilated_conv3d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  arr ->
  int array ->
  int array ->
  arr

Source

val transpose_conv1d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  arr ->
  int array ->
  arr

Source

val transpose_conv2d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  arr ->
  int array ->
  arr

Source

val transpose_conv3d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  arr ->
  int array ->
  arr

Source

val max_pool1d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr

Source

val max_pool2d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr

Source

val max_pool3d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr

Source

val avg_pool1d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr

Source

val avg_pool2d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr

Source

val avg_pool3d : 
  ?padding:Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr

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

Source

val dilated_conv1d_backward_input : 
  arr ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val dilated_conv1d_backward_kernel : 
  arr ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val dilated_conv2d_backward_input : 
  arr ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val dilated_conv2d_backward_kernel : 
  arr ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val dilated_conv3d_backward_input : 
  arr ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val 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

Source

val max_pool1d_backward : 
  Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val max_pool2d_backward : 
  Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val max_pool3d_backward : 
  Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val avg_pool1d_backward : 
  Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val avg_pool2d_backward : 
  Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Source

val avg_pool3d_backward : 
  Owl_types_common.padding ->
  arr ->
  int array ->
  int array ->
  arr ->
  arr

Sourceval upsampling2d_backward : arr -> int array -> arr -> arr

Sourceval row_num : arr -> int

Sourceval col_num : arr -> int

Sourceval row : arr -> int -> arr

Sourceval rows : arr -> int array -> arr

Sourceval copy_row_to : arr -> arr -> int -> unit

Sourceval copy_col_to : arr -> arr -> int -> unit

Sourceval dot : arr -> arr -> arr

Sourceval inv : arr -> arr

Sourceval logdet : arr -> elt

Sourceval chol : ?upper:bool -> arr -> arr

Sourceval svd : ?thin:bool -> arr -> arr * arr * arr

Sourceval qr : arr -> arr * arr

Sourceval lyapunov : arr -> arr -> arr

Source

val discrete_lyapunov : 
  ?solver:[ `bilinear | `default | `direct ] ->
  arr ->
  arr ->
  arr

Sourceval diag : ?k:int -> arr -> arr

Sourceval diagm : ?k:int -> arr -> arr

Sourceval triu : ?k:int -> arr -> arr

Sourceval tril : ?k:int -> arr -> arr

Sourceval trace : arr -> elt

Sourceval transpose : ?axis:int array -> 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 of_array : elt array -> int array -> arr

Sourceval of_arrays : elt array array -> arr

Sourceval float_to_elt : float -> elt

Sourceval elt_to_float : elt -> float

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 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

Sourcemodule Scalar : sig ... end

package owl

Module Algodiff.ASource

Module `Algodiff.A`Source