owl 0.2.9 · OCaml Package

type arr = Owl_dense_ndarray.S.arr

type mat = Owl_dense_matrix.S.mat

type elt = Owl_dense_matrix.S.elt

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

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

| F of float
| Arr of arr
| Mat of mat
| DF of t * t * int
| DR of t * t Pervasives.ref * trace_op * int Pervasives.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 pack_mat : mat -> t

val unpack_mat : t -> mat

val tag : unit -> int

val primal : t -> t

val primal' : t -> t

val adjval : t -> t

val adjref : t -> t Pervasives.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 clone_primal' : t -> t

val clip_by_l2norm : elt -> t -> t

val to_trace : t list -> string

val to_dot : t list -> string

module Utils : sig ... end

module Learning_Rate : sig ... end

module Batch : sig ... end

module Loss : sig ... end

module Gradient : sig ... end

module Momentum : sig ... end

module Regularisation : sig ... end

module Clipping : sig ... end

module Stopping : sig ... end

module Checkpoint : sig ... end

module Params : sig ... end

val minimise_weight : 
  ?state:Checkpoint.state ->
  Params.typ ->
  (t -> t -> t) ->
  t ->
  t ->
  t ->
  Checkpoint.state * t

val minimise_network : 
  ?state:Checkpoint.state ->
  Params.typ ->
  (t -> t * t array array) ->
  (t -> t array array * t array array) ->
  (t array array -> 'a) ->
  (string -> unit) ->
  t ->
  t ->
  Checkpoint.state

package owl