owl 0.3.0 · OCaml Package

owl

type arr = Owl_dense_ndarray.S.arr

type elt = Owl_dense_ndarray.S.elt

type trace_op = Owl_algodiff_generic.Make(Owl_dense_ndarray.S).trace_op

type t = Owl_algodiff_generic.Make(Owl_dense_ndarray.S).t =

| F of float
| Arr of arr
| DF of t * t * int
| DR of t * t Stdlib.ref * trace_op * int Stdlib.ref * int

module Maths : sig ... end

module Mat : sig ... end

module Arr : sig ... end

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

val pack_flt : elt -> t

val unpack_flt : t -> elt

val pack_arr : arr -> t

val unpack_arr : t -> arr

val tag : unit -> int

val primal : t -> t

val primal' : t -> t

val adjval : t -> t

val adjref : t -> t Stdlib.ref

val tangent : t -> t

val make_forward : t -> t -> int -> t

val make_reverse : t -> int -> t

val reverse_prop : t -> t -> unit

val type_info : t -> string

val shape : t -> int array

val copy_primal' : t -> t

val clip_by_l2norm : elt -> t -> t

val to_trace : t list -> string

val to_dot : t list -> string

val pp_num : Stdlib.Format.formatter -> t -> unit

module Init : sig ... end

module Input : sig ... end

module Activation : sig ... end

module Linear : sig ... end

module LinearNoBias : sig ... end

module Recurrent : sig ... end

module LSTM : sig ... end

module GRU : sig ... end

module Conv1D : sig ... end

module Conv2D : sig ... end

module Conv3D : sig ... end

module FullyConnected : sig ... end

module MaxPool1D : sig ... end

module MaxPool2D : sig ... end

module AvgPool1D : sig ... end

module AvgPool2D : sig ... end

module GlobalMaxPool1D : sig ... end

module GlobalMaxPool2D : sig ... end

module GlobalAvgPool1D : sig ... end

module GlobalAvgPool2D : sig ... end

module UpSampling1D : sig ... end

module UpSampling2D : sig ... end

module UpSampling3D : sig ... end

module Padding1D : sig ... end

module Padding2D : sig ... end

module Padding3D : sig ... end

module Lambda : sig ... end

module Dropout : sig ... end

module Reshape : sig ... end

module Flatten : sig ... end

module Add : sig ... end

module Mul : sig ... end

module Dot : sig ... end

module Max : sig ... end

module Average : sig ... end

module Concatenate : sig ... end

module Normalisation : sig ... end

module GaussianNoise : sig ... end

module GaussianDropout : sig ... end

module AlphaDropout : sig ... end

module Embedding : sig ... end

module Masking : sig ... end

type neuron = Owl_neural_neuron.Make(Owl_dense_ndarray.S).neuron =

| Input of Input.neuron_typ
| Linear of Linear.neuron_typ
| LinearNoBias of LinearNoBias.neuron_typ
| Embedding of Embedding.neuron_typ
| LSTM of LSTM.neuron_typ
| GRU of GRU.neuron_typ
| Recurrent of Recurrent.neuron_typ
| Conv1D of Conv1D.neuron_typ
| Conv2D of Conv2D.neuron_typ
| Conv3D of Conv3D.neuron_typ
| FullyConnected of FullyConnected.neuron_typ
| MaxPool1D of MaxPool1D.neuron_typ
| MaxPool2D of MaxPool2D.neuron_typ
| AvgPool1D of AvgPool1D.neuron_typ
| AvgPool2D of AvgPool2D.neuron_typ
| GlobalMaxPool1D of GlobalMaxPool1D.neuron_typ
| GlobalMaxPool2D of GlobalMaxPool2D.neuron_typ
| GlobalAvgPool1D of GlobalAvgPool1D.neuron_typ
| GlobalAvgPool2D of GlobalAvgPool2D.neuron_typ
| Dropout of Dropout.neuron_typ
| Reshape of Reshape.neuron_typ
| Flatten of Flatten.neuron_typ
| Lambda of Lambda.neuron_typ
| Activation of Activation.neuron_typ
| GaussianNoise of GaussianNoise.neuron_typ
| GaussianDropout of GaussianDropout.neuron_typ
| AlphaDropout of AlphaDropout.neuron_typ
| Normalisation of Normalisation.neuron_typ
| Add of Add.neuron_typ
| Mul of Mul.neuron_typ
| Dot of Dot.neuron_typ
| Max of Max.neuron_typ
| Average of Average.neuron_typ
| Concatenate of Concatenate.neuron_typ

val get_in_out_shape : neuron -> int array * int array

val get_in_shape : neuron -> int array

val get_out_shape : neuron -> int array

val connect : int array array -> neuron -> unit

val init : neuron -> unit

val reset : neuron -> unit

val mktag : int -> neuron -> unit

val mkpar : neuron -> t array

val mkpri : neuron -> t array

val mkadj : neuron -> t array

val update : neuron -> t array -> unit

val copy : neuron -> neuron

val run : t array -> neuron -> t

val to_string : neuron -> string

val to_name : neuron -> string

package owl