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
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. | Add_Row_D_D of t * t * int
  69. | Add_Row_D_C of t * t * int
  70. | Add_Row_C_D of t * t * int
  71. | Get_Row_D of t * int
  72. | Of_Rows_D of t array
  73. | Concat_D_D of t * t * int
  74. | Concat_D_C of t * t * int
  75. | Concat_C_D of t * t * int
  76. | Conv1D_D_D of t * t * int array
  77. | Conv1D_D_C of t * t * int array
  78. | Conv1D_C_D of t * t * int array
  79. | Conv2D_D_D of t * t * int array
  80. | Conv2D_D_C of t * t * int array
  81. | Conv2D_C_D of t * t * int array
  82. | Conv3D_D_D of t * t * int array
  83. | Conv3D_D_C of t * t * int array
  84. | Conv3D_C_D of t * t * int array
  85. | Di_Conv1D_D_D of t * t * int array * int array
  86. | Di_Conv1D_D_C of t * t * int array * int array
  87. | Di_Conv1D_C_D of t * t * int array * int array
  88. | Di_Conv2D_D_D of t * t * int array * int array
  89. | Di_Conv2D_D_C of t * t * int array * int array
  90. | Di_Conv2D_C_D of t * t * int array * int array
  91. | Di_Conv3D_D_D of t * t * int array * int array
  92. | Di_Conv3D_D_C of t * t * int array * int array
  93. | Di_Conv3D_C_D of t * t * int array * int array
  94. | Tr_Conv1D_D_D of t * t * int array
  95. | Tr_Conv1D_D_C of t * t * int array
  96. | Tr_Conv1D_C_D of t * t * int array
  97. | Tr_Conv2D_D_D of t * t * int array
  98. | Tr_Conv2D_D_C of t * t * int array
  99. | Tr_Conv2D_C_D of t * t * int array
  100. | Tr_Conv3D_D_D of t * t * int array
  101. | Tr_Conv3D_D_C of t * t * int array
  102. | Tr_Conv3D_C_D of t * t * int array
  103. | Reshape_D of t
  104. | Maxpool1D_D of t * Owl_types.padding * int array * int array
  105. | Maxpool2D_D of t * Owl_types.padding * int array * int array
  106. | Maxpool3D_D of t * Owl_types.padding * int array * int array
  107. | Avgpool1D_D of t * Owl_types.padding * int array * int array
  108. | Avgpool2D_D of t * Owl_types.padding * int array * int array
  109. | Avgpool3D_D of t * Owl_types.padding * int array * int array
val _global_tag : int ref
val tag : unit -> int
val cmp_tag : 'a -> 'a -> 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
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