package owl-base

  1. Overview
  2. Docs

Module Optimise.Algodiff

module A : sig ... end
and adjoint = t -> t ref -> (t * t) list -> (t * t) list
and register = t list -> t list
and label = string * t list
and op = adjoint * register * label
val tag : unit -> int
val primal : t -> t
val primal' : t -> t
val zero : t -> t
val reset_zero : t -> t
val tangent : t -> t
val adjref : t -> t ref
val adjval : t -> t
val shape : t -> int array
val row_num : t -> int
val col_num : t -> int
val numel : t -> int
val clip_by_value : amin:A.elt -> amax:A.elt -> t -> t
val clip_by_l2norm : A.elt -> t -> t
val copy_primal' : t -> t
val tile : t -> int array -> t
val repeat : t -> int array -> t
val pack_elt : A.elt -> t
val unpack_elt : t -> A.elt
val pack_flt : float -> t
val _f : float -> t
val unpack_flt : t -> float
val pack_arr : A.arr -> t
val unpack_arr : t -> A.arr
val deep_info : t -> string
val type_info : t -> string
val error_binop : string -> t -> t -> 'a
val error_uniop : string -> t -> 'a
val make_forward : t -> t -> int -> t
val make_reverse : t -> int -> t
val reverse_prop : t -> t -> unit
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
module Builder : sig ... end
module Maths : sig ... end
module Linalg : sig ... end
module NN : sig ... end
module Mat : sig ... end
module Arr : sig ... end
val to_trace : t list -> string
val to_dot : t list -> string
val pp_num : Format.formatter -> t -> unit