package owl-base

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

Module Optimiser.OperatorSource

Vectorised functions
Sourceval empty : int array -> Symbol.Shape.Type.arr

TODO

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

TODO

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

TODO

TODO

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

TODO

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

TODO

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

TODO

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

TODO

Sourceval init_nd : int array -> (int array -> Symbol.Shape.Type.elt) -> Symbol.Shape.Type.arr

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

TODO

TODO

TODO

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

TODO

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

TODO

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

TODO

``delay f x`` returns ``f x``. It allows to use a function that is not tracked by the computation graph and delay its evaluation. The output must have the same shape as the input.

``delay_array out_shape f x`` works in the same way as ``delay`` but is applied on an array of ndarrays. Needs the shape of the output as an argument.

Sourceval 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

``lazy_print x`` prints the output of ``x`` when it is evaluated. Is implemented as an identity node. For information about the optional parameters, refer to the ``print`` function of the ``Ndarray`` module.

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

TODO

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

TODO

TODO

TODO

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

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

TODO

Sourceval max_pool1d_backward : Owl_types.padding -> Symbol.Shape.Type.arr -> int array -> int array -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

TODO

Sourceval max_pool2d_backward : Owl_types.padding -> Symbol.Shape.Type.arr -> int array -> int array -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

TODO

Sourceval max_pool3d_backward : Owl_types.padding -> Symbol.Shape.Type.arr -> int array -> int array -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

TODO

Sourceval avg_pool1d_backward : Owl_types.padding -> Symbol.Shape.Type.arr -> int array -> int array -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

TODO

Sourceval avg_pool2d_backward : Owl_types.padding -> Symbol.Shape.Type.arr -> int array -> int array -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

TODO

Sourceval avg_pool3d_backward : Owl_types.padding -> Symbol.Shape.Type.arr -> int array -> int array -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

TODO

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

TODO

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

TODO

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

TODO

TODO

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

TODO

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

TODO

TODO

Sourceval discrete_lyapunov : ?solver:[ `default | `bilinear | `direct ] -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr -> Symbol.Shape.Type.arr

TODO

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

TODO

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

TODO

Sourceval to_cols : Symbol.Shape.Type.arr -> 'a array

TODO

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

TODO

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

TODO

Sourceval to_arrays : Symbol.Shape.Type.arr -> Symbol.Shape.Type.elt array array

TODO

Scalar functions
Sourcemodule Scalar : sig ... end