owl 0.4.0 · OCaml Package

val noop : 'a -> unit

val op_d_d : 
  t ->
  (t -> t) ->
  (t -> t) ->
  (t -> t -> t -> t) ->
  (t -> trace_op) ->
  t

val op_d_d_d : 
  t ->
  t ->
  (t -> t -> t) ->
  (t -> t -> t) ->
  (t -> t -> t -> t) ->
  (t -> t -> t -> t) ->
  (t -> t -> t -> t -> t -> t) ->
  (t -> t -> trace_op) ->
  (t -> t -> trace_op) ->
  (t -> t -> trace_op) ->
  t

val (+) : t -> t -> t

val add : t -> t -> t

val (-) : t -> t -> t

val sub : t -> t -> t

val (*) : t -> t -> t

val mul : t -> t -> t

val (/) : t -> t -> t

val div : t -> t -> t

val (**) : t -> t -> t

val pow : t -> t -> t

val atan2 : t -> t -> t

val min2 : t -> t -> t

val max2 : t -> t -> t

val neg : t -> t

val abs : t -> t

val signum : t -> t

val floor : t -> t

val ceil : t -> t

val round : t -> t

val sqr : t -> t

val sqrt : t -> t

val log : t -> t

val log2 : t -> t

val log10 : t -> t

val exp : t -> t

val sin : t -> t

val cos : t -> t

val tan : t -> t

val sinh : t -> t

val cosh : t -> t

val tanh : t -> t

val asin : t -> t

val acos : t -> t

val atan : t -> t

val asinh : t -> t

val acosh : t -> t

val atanh : t -> t

val get_item : t -> int -> int -> t

val set_item : t -> int -> int -> t -> t

val add_item : t -> int -> int -> t -> t

val get_slice : int list list -> t -> t

val set_slice : int list list -> t -> t -> t

val sum' : t -> t

val sum : ?axis:int -> t -> t

val sum_reduce : ?axis:int array -> t -> t

val mean : t -> t

val (*@) : t -> t -> t

val dot : t -> t -> t

val transpose : t -> t

val l1norm' : t -> t

val l2norm' : t -> t

val l2norm_sqr' : t -> t

val sigmoid : t -> t

val relu : t -> t

val inv : t -> t

val softplus : t -> t

val softsign : t -> t

val softmax : ?axis:int -> t -> t

val cross_entropy : t -> t -> t

val add_row : t -> t -> int -> t

val get_row : t -> int -> t

val to_rows : t -> t array

val of_rows : t array -> t

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

val conv1d_backward_input : t -> t -> int array -> t -> t

val conv1d_backward_kernel : t -> t -> int array -> t -> t

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

val conv2d_backward_input : t -> t -> int array -> t -> t

val conv2d_backward_kernel : t -> t -> int array -> t -> t

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

val conv3d_backward_input : t -> t -> int array -> t -> t

val conv3d_backward_kernel : t -> t -> int array -> t -> t

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

val dilated_conv1d_backward_input : t -> t -> int array -> int array -> t -> t

val dilated_conv1d_backward_kernel : t -> t -> int array -> int array -> t -> t

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

val dilated_conv2d_backward_input : t -> t -> int array -> int array -> t -> t

val dilated_conv2d_backward_kernel : t -> t -> int array -> int array -> t -> t

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

val dilated_conv3d_backward_input : t -> t -> int array -> int array -> t -> t

val dilated_conv3d_backward_kernel : t -> t -> int array -> int array -> t -> t

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

val transpose_conv1d_backward_input : t -> t -> int array -> t -> t

val transpose_conv1d_backward_kernel : t -> t -> int array -> t -> t

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

val transpose_conv2d_backward_input : t -> t -> int array -> t -> t

val transpose_conv2d_backward_kernel : t -> t -> int array -> t -> t

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

val transpose_conv3d_backward_input : t -> t -> int array -> t -> t

val transpose_conv3d_backward_kernel : t -> t -> int array -> t -> t

val reshape : t -> int array -> t

val flatten : t -> t

val max_pool1d : Owl_types.padding -> t -> int array -> int array -> t

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

val max_pool2d : Owl_types.padding -> t -> int array -> int array -> t

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

val max_pool3d : Owl_types.padding -> t -> int array -> int array -> t

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

val avg_pool1d : Owl_types.padding -> t -> int array -> int array -> t

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

val avg_pool2d : Owl_types.padding -> t -> int array -> int array -> t

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

val avg_pool3d : Owl_types.padding -> t -> int array -> int array -> t

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

val dropout : ?rate:float -> t -> t

val concat : int -> t -> t -> t

val split : int -> int array -> t -> t array

package owl