package owl-base

  1. Overview
  2. Docs
package owl-base
Legend:
Library
Module
Module type
Parameter
Class
Class type

Parameters

Signature

module A = A
type t =
  1. | F of A.elt
  2. | Arr of A.arr
  3. | DF of t * t * int
  4. | DR of t * t ref * trace_op * int ref * int * int ref
and trace_op =
  1. | Noop
  2. | Add_D_D of t * t
  3. | Add_D_C of t * t
  4. | Add_C_D of t * t
  5. | Sub_D_D of t * t
  6. | Sub_D_C of t * t
  7. | Sub_C_D of t * t
  8. | Mul_D_D of t * t
  9. | Mul_D_C of t * t
  10. | Mul_C_D of t * t
  11. | Div_D_D of t * t
  12. | Div_D_C of t * t
  13. | Div_C_D of t * t
  14. | Pow_D_D of t * t
  15. | Pow_D_C of t * t
  16. | Pow_C_D of t * t
  17. | Atan2_D_D of t * t
  18. | Atan2_D_C of t * t
  19. | Atan2_C_D of t * t
  20. | Neg_D of t
  21. | Abs_D of t
  22. | Signum_D of t
  23. | Floor_D of t
  24. | Ceil_D of t
  25. | Round_D of t
  26. | Sqr_D of t
  27. | Sqrt_D of t
  28. | Log_D of t
  29. | Log2_D of t
  30. | Log10_D of t
  31. | Exp_D of t
  32. | Sin_D of t
  33. | Cos_D of t
  34. | Tan_D of t
  35. | Sinh_D of t
  36. | Cosh_D of t
  37. | Tanh_D of t
  38. | Asin_D of t
  39. | Acos_D of t
  40. | Atan_D of t
  41. | Asinh_D of t
  42. | Acosh_D of t
  43. | Atanh_D of t
  44. | Get_Item of t * int * int
  45. | SetI_D_D of t * int * int * t
  46. | SetI_D_C of t * int * int * t
  47. | SetI_C_D of t * int * int * t
  48. | AddI_D_D of t * int * int * t
  49. | AddI_D_C of t * int * int * t
  50. | AddI_C_D of t * int * int * t
  51. | Get_Slice_D of t * int list list
  52. | Set_Slice_D_D of t * t * int list list
  53. | Set_Slice_D_C of t * t * int list list
  54. | Set_Slice_C_D of t * t * int list list
  55. | Sum_D of t
  56. | Sum__D of t * int
  57. | Sum___D of t * int array
  58. | Dot_D_D of t * t
  59. | Dot_D_C of t * t
  60. | Dot_C_D of t * t
  61. | Trans_D of t
  62. | L1Norm_D of t
  63. | L2Norm_D of t
  64. | L2NormS_D of t
  65. | Sigmoid_D of t
  66. | Relu_D of t
  67. | Inv_D of t
  68. | Logdet_D of t
  69. | Chol_D of t * bool
  70. | QR_D of t * (t ref * t ref) * (t ref * t ref)
  71. | Svd_D of t * (t ref * t ref * t ref) * (t ref * t ref * t ref) * bool
  72. | Lyapunov_D_D of t * t
  73. | Lyapunov_C_D of t * t
  74. | Lyapunov_D_C of t * t
  75. | Discrete_Lyapunov_D_D of t * t
  76. | Discrete_Lyapunov_C_D of t * t
  77. | Discrete_Lyapunov_D_C of t * t
  78. | Diag_D of int * t
  79. | Diagm_D of int * t
  80. | Tril_D of int * t
  81. | Triu_D of int * t
  82. | Trace_D of t
  83. | Add_Row_D_D of t * t * int
  84. | Add_Row_D_C of t * t * int
  85. | Add_Row_C_D of t * t * int
  86. | Get_Row_D of t * int
  87. | Of_Rows_D of t array
  88. | Of_Arrays_D of t array array * (int * int) list
  89. | Concat_D_D of t * t * int
  90. | Concat_D_C of t * t * int
  91. | Concat_C_D of t * t * int
  92. | Split_D of t * int * t ref array
  93. | Concatenate_D of t array * int * int list
  94. | Conv1D_D_D of t * t * int array
  95. | Conv1D_D_C of t * t * int array
  96. | Conv1D_C_D of t * t * int array
  97. | Conv2D_D_D of t * t * int array
  98. | Conv2D_D_C of t * t * int array
  99. | Conv2D_C_D of t * t * int array
  100. | Conv3D_D_D of t * t * int array
  101. | Conv3D_D_C of t * t * int array
  102. | Conv3D_C_D of t * t * int array
  103. | Di_Conv1D_D_D of t * t * int array * int array
  104. | Di_Conv1D_D_C of t * t * int array * int array
  105. | Di_Conv1D_C_D of t * t * int array * int array
  106. | Di_Conv2D_D_D of t * t * int array * int array
  107. | Di_Conv2D_D_C of t * t * int array * int array
  108. | Di_Conv2D_C_D of t * t * int array * int array
  109. | Di_Conv3D_D_D of t * t * int array * int array
  110. | Di_Conv3D_D_C of t * t * int array * int array
  111. | Di_Conv3D_C_D of t * t * int array * int array
  112. | Tr_Conv1D_D_D of t * t * int array
  113. | Tr_Conv1D_D_C of t * t * int array
  114. | Tr_Conv1D_C_D of t * t * int array
  115. | Tr_Conv2D_D_D of t * t * int array
  116. | Tr_Conv2D_D_C of t * t * int array
  117. | Tr_Conv2D_C_D of t * t * int array
  118. | Tr_Conv3D_D_D of t * t * int array
  119. | Tr_Conv3D_D_C of t * t * int array
  120. | Tr_Conv3D_C_D of t * t * int array
  121. | Reshape_D of t
  122. | Maxpool1D_D of t * Owl_types.padding * int array * int array
  123. | Maxpool2D_D of t * Owl_types.padding * int array * int array
  124. | Maxpool3D_D of t * Owl_types.padding * int array * int array
  125. | Avgpool1D_D of t * Owl_types.padding * int array * int array
  126. | Avgpool2D_D of t * Owl_types.padding * int array * int array
  127. | Avgpool3D_D of t * Owl_types.padding * int array * int array
  128. | UpSampling2D_D of t * int array
  129. | PAD_D of t * int list list
val _global_tag : int ref
val tag : unit -> int
val cmp_tag : 'a -> 'b -> int
val reset_zero : t -> t
val primal : t -> t
val primal' : t -> t
val 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
module Maths : sig ... end
val reverse_reset : t -> unit
val _shrink : t -> t -> t
val reverse_push : t -> t -> unit
val reverse_prop : t -> t -> unit
val make_forward : t -> t -> int -> t
val make_reverse : t -> int -> t
val diff' : (t -> t) -> t -> t * t
val diff : (t -> t) -> t -> t
val grad' : (t -> t) -> t -> t * t
val grad : (t -> t) -> t -> t
val jacobianv' : (t -> t) -> t -> t -> t * t
val jacobianv : (t -> t) -> t -> t -> t
val jacobianTv' : (t -> t) -> t -> t -> t * t
val jacobianTv : (t -> t) -> t -> t -> t
val jacobian' : (t -> t) -> t -> t * t
val jacobian : (t -> t) -> t -> t
val gradhessian : (t -> t) -> t -> t * t
val gradhessian' : (t -> t) -> t -> t * t * t
val hessian : (t -> t) -> t -> t
val hessian' : (t -> t) -> t -> t * t
val gradhessianv' : (t -> t) -> t -> t -> t * t * t
val gradhessianv : (t -> t) -> t -> t -> t * t
val hessianv' : (t -> t) -> t -> t -> t * t
val hessianv : (t -> t) -> t -> t -> t
val laplacian : (t -> t) -> t -> t
val laplacian' : (t -> t) -> t -> t * t
module Mat : sig ... end
module Arr : sig ... end
val _traverse_trace : t list -> (t, int * string * t list) Hashtbl.t
val _convert_terminal_output : (t, int * string * t list) Hashtbl.t -> string
val _convert_dot_output : (t, int * string * t list) Hashtbl.t -> string
val to_trace : t list -> string
val to_dot : t list -> string
val pp_num : Format.formatter -> t -> unit
module FDGrad_test : sig ... end
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct