Make (owl-base.Owl_computation

Parameters

module Symbol : Owl_computation_symbol_sig.Sig

Signature

module Symbol = Symbol

val noop : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val empty : int array -> Symbol.Shape.Type.arr

val zeros : int array -> Symbol.Shape.Type.arr

val ones : int array -> Symbol.Shape.Type.arr

val create : int array -> Symbol.Shape.Type.elt -> Symbol.Shape.Type.arr

val sequential : 
  ?a:Symbol.Shape.Type.elt ->
  ?step:Symbol.Shape.Type.elt ->
  int array ->
  Symbol.Shape.Type.arr

val uniform : 
  ?a:Symbol.Shape.Type.elt ->
  ?b:Symbol.Shape.Type.elt ->
  int array ->
  Symbol.Shape.Type.arr

val gaussian : 
  ?mu:Symbol.Shape.Type.elt ->
  ?sigma:Symbol.Shape.Type.elt ->
  int array ->
  Symbol.Shape.Type.arr

val bernoulli : ?p:Symbol.Shape.Type.elt -> int array -> Symbol.Shape.Type.arr

val init : int array -> (int -> Symbol.Shape.Type.elt) -> Symbol.Shape.Type.arr

val init_nd : 'a -> 'b -> 'c

val shape : Symbol.Shape.Type.arr -> int array

val numel : Symbol.Shape.Type.arr -> int

val get : Symbol.Shape.Type.arr -> int array -> Symbol.Shape.Type.elt

val set : Symbol.Shape.Type.arr -> int array -> Symbol.Shape.Type.elt -> unit

val get_slice : int list list -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val set_slice : 
  int list list ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  unit

val get_fancy : 
  Owl_types_common.index list ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val set_fancy : 
  Owl_types_common.index list ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  unit

val copy : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val copy_ : out:'a -> 'b -> 'c

val reset : Symbol.Shape.Type.arr -> unit

val reshape : Symbol.Shape.Type.arr -> int array -> Symbol.Shape.Type.arr

val reverse : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val tile : Symbol.Shape.Type.arr -> int array -> Symbol.Shape.Type.arr

val repeat : Symbol.Shape.Type.arr -> int array -> Symbol.Shape.Type.arr

val pad : 
  ?v:Symbol.Shape.Type.elt ->
  int list list ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val expand : ?hi:bool -> 'a -> 'b -> 'c

val squeeze : ?axis:'a array -> 'b -> 'c

val concatenate : 
  ?axis:int ->
  Symbol.Shape.Type.arr array ->
  Symbol.Shape.Type.arr

val stack : ?axis:int -> Symbol.Shape.Type.arr array -> Symbol.Shape.Type.arr

val concat : axis:'a -> 'b

val split : ?axis:int -> 'a -> 'b -> 'c

val draw : 
  ?axis:int ->
  Symbol.Shape.Type.arr ->
  int ->
  Symbol.Shape.Type.arr * 'a array

val map : 
  (Symbol.Shape.Type.elt -> Symbol.Shape.Type.elt) ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val fold : 
  ?axis:int ->
  (Symbol.Shape.Type.elt -> Symbol.Shape.Type.elt -> Symbol.Shape.Type.elt) ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val scan : 
  ?axis:int ->
  (Symbol.Shape.Type.elt -> Symbol.Shape.Type.elt -> Symbol.Shape.Type.elt) ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val one_hot : int -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val delay : 
  (Symbol.Shape.Type.Device.A.arr -> Symbol.Shape.Type.Device.A.arr) ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val delay_array : 
  int array ->
  (Symbol.Shape.Type.Device.A.arr array -> Symbol.Shape.Type.Device.A.arr) ->
  Symbol.Shape.Type.arr array ->
  Symbol.Shape.Type.arr

val lazy_print : 
  ?max_row:int ->
  ?max_col:int ->
  ?header:bool ->
  ?fmt:(Symbol.Shape.Type.Device.A.elt -> string) ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val print : ?max_row:'a -> ?max_col:'b -> ?header:'c -> ?fmt:'d -> 'e -> unit

val abs : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val neg : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val floor : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val ceil : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val round : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val sqr : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val sqrt : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val log : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val log2 : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val log10 : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val exp : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val sin : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val cos : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val tan : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val sinh : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val cosh : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val tanh : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val asin : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val acos : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val atan : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val asinh : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val acosh : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val atanh : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val min : 
  ?axis:int ->
  ?keep_dims:bool ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val max : 
  ?axis:int ->
  ?keep_dims:bool ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val sum : 
  ?axis:int ->
  ?keep_dims:bool ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val sum_reduce : 
  ?axis:int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val signum : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val sigmoid : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val relu : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val dawsn : Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

val min' : Symbol.Shape.Type.arr -> Symbol.Shape.Type.elt

val max' : Symbol.Shape.Type.arr -> Symbol.Shape.Type.elt

val sum' : Symbol.Shape.Type.arr -> Symbol.Shape.Type.elt

val log_sum_exp' : Symbol.Shape.Type.arr -> Symbol.Shape.Type.elt

val log_sum_exp : 
  ?axis:int ->
  ?keep_dims:bool ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val l1norm' : Symbol.Shape.Type.arr -> Symbol.Shape.Type.elt

val l2norm' : Symbol.Shape.Type.arr -> Symbol.Shape.Type.elt

val l2norm_sqr' : Symbol.Shape.Type.arr -> Symbol.Shape.Type.elt

val clip_by_value : 
  ?amin:Symbol.Shape.Type.elt ->
  ?amax:Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val clip_by_l2norm : 
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val pow : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val scalar_pow : 
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val pow_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val atan2 : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val scalar_atan2 : 
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val atan2_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val hypot : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val min2 : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val max2 : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val add : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val sub : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val mul : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val div : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val add_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val sub_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val mul_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val div_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val scalar_add : 
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val scalar_sub : 
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val scalar_mul : 
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val scalar_div : 
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val fma : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val elt_equal : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val elt_not_equal : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val elt_less : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val elt_greater : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val elt_less_equal : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val elt_greater_equal : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val elt_equal_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val elt_not_equal_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val elt_less_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val elt_greater_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val elt_less_equal_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val elt_greater_equal_scalar : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.elt ->
  Symbol.Shape.Type.arr

val conv1d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr

val conv2d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr

val conv3d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr

val transpose_conv1d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr

val transpose_conv2d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr

val transpose_conv3d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr

val dilated_conv1d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr

val dilated_conv2d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr

val dilated_conv3d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr

val max_pool1d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr

val max_pool2d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr

val max_pool3d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr

val avg_pool1d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr

val avg_pool2d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr

val avg_pool3d : 
  ?padding:Owl_types.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr

val upsampling2d : Symbol.Shape.Type.arr -> int array -> Symbol.Shape.Type.arr

val conv1d_backward_input : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val conv1d_backward_kernel : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val conv2d_backward_input : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val conv2d_backward_kernel : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val conv3d_backward_input : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val conv3d_backward_kernel : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val transpose_conv1d_backward_input : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val transpose_conv1d_backward_kernel : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val transpose_conv2d_backward_input : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val transpose_conv2d_backward_kernel : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val transpose_conv3d_backward_input : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val transpose_conv3d_backward_kernel : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val dilated_conv1d_backward_input : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val dilated_conv1d_backward_kernel : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val dilated_conv2d_backward_input : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val dilated_conv2d_backward_kernel : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val dilated_conv3d_backward_input : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val dilated_conv3d_backward_kernel : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val max_pool1d_backward : 
  Owl_types_common.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val max_pool2d_backward : 
  Owl_types_common.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val max_pool3d_backward : 
  Owl_types_common.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val avg_pool1d_backward : 
  Owl_types_common.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val avg_pool2d_backward : 
  Owl_types_common.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val avg_pool3d_backward : 
  Owl_types_common.padding ->
  Symbol.Shape.Type.arr ->
  int array ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val upsampling2d_backward : 
  Symbol.Shape.Type.arr ->
  int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val row_num : Symbol.Shape.Type.arr -> int

val col_num : Symbol.Shape.Type.arr -> int

val row : Symbol.Shape.Type.arr -> 'a -> Symbol.Shape.Type.arr

val rows : Symbol.Shape.Type.arr -> int array -> Symbol.Shape.Type.arr

val copy_row_to : Symbol.Shape.Type.arr -> 'a -> 'b -> unit

val copy_col_to : Symbol.Shape.Type.arr -> 'a -> 'b -> unit

val trace : Symbol.Shape.Type.arr -> Symbol.Shape.Type.elt

val diag : ?k:'a -> 'b -> 'c

val dot : 
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val transpose : 
  ?axis:int array ->
  Symbol.Shape.Type.arr ->
  Symbol.Shape.Type.arr

val to_rows : Symbol.Shape.Type.arr -> 'a array

val of_rows : Symbol.Shape.Type.arr array -> Symbol.Shape.Type.arr

val of_array : 
  Symbol.Shape.Type.elt array ->
  int array ->
  Symbol.Shape.Type.arr

val of_cols : 'a -> 'b

val to_cols : 'a -> 'b

val of_arrays : Symbol.Shape.Type.elt array array -> Symbol.Shape.Type.arr

val to_arrays : 'a -> 'b

Scalar maths

module Scalar : sig ... end

module Mat : sig ... end

module Linalg : sig ... end