package owl-base

  1. Overview
  2. Docs

Module Owl_optimise_generic.MakeSource

Parameters

Signature

include sig ... end
Sourcetype arr = A.arr
Sourcetype elt = A.elt
Sourcetype t = Owl_algodiff_generic.Make(A).t =
  1. | F of float
  2. | Arr of arr
  3. | DF of t * t * int
  4. | DR of t * t ref * trace_op * int ref * int
Sourceand trace_op = Owl_algodiff_generic.Make(A).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. | Dot_D_D of t * t
  58. | Dot_D_C of t * t
  59. | Dot_C_D of t * t
  60. | Trans_D of t
  61. | L1Norm_D of t
  62. | L2Norm_D of t
  63. | L2NormS_D of t
  64. | Sigmoid_D of t
  65. | Relu_D of t
  66. | Inv_D of t
  67. | Add_Row_D_D of t * t * int
  68. | Add_Row_D_C of t * t * int
  69. | Add_Row_C_D of t * t * int
  70. | Get_Row_D of t * int
  71. | Of_Rows_D of t array
  72. | Concat_D_D of t * t * int
  73. | Concat_D_C of t * t * int
  74. | Concat_C_D of t * t * int
  75. | Conv1D_D_D of t * t * int array
  76. | Conv1D_D_C of t * t * int array
  77. | Conv1D_C_D of t * t * int array
  78. | Conv2D_D_D of t * t * int array
  79. | Conv2D_D_C of t * t * int array
  80. | Conv2D_C_D of t * t * int array
  81. | Conv3D_D_D of t * t * int array
  82. | Conv3D_D_C of t * t * int array
  83. | Conv3D_C_D of t * t * int array
  84. | Reshape_D of t
  85. | Maxpool1D_D of t * Owl_types.padding * int array * int array
  86. | Maxpool2D_D of t * Owl_types.padding * int array * int array
  87. | Maxpool3D_D of t * Owl_types.padding * int array * int array
  88. | Avgpool1D_D of t * Owl_types.padding * int array * int array
  89. | Avgpool2D_D of t * Owl_types.padding * int array * int array
  90. | Avgpool3D_D of t * Owl_types.padding * int array * int array
Sourceval _global_tag : int ref
Sourceval tag : unit -> int
Sourceval cmp_tag : 'a -> 'a -> int
Sourceval reset_zero : t -> t
Sourceval primal : t -> t
Sourceval primal' : t -> t
Sourceval zero : t -> t
Sourceval tangent : t -> t
Sourceval adjref : t -> t ref
Sourceval adjval : t -> t
Sourceval shape : t -> int array
Sourceval row_num : t -> int
Sourceval col_num : t -> int
Sourceval numel : t -> int
Sourceval clip_by_value : amin:A.elt -> amax:A.elt -> t -> t
Sourceval clip_by_l2norm : A.elt -> t -> t
Sourceval copy_primal' : t -> t
Sourceval tile : t -> int array -> t
Sourceval repeat : ?axis:int -> t -> int -> t
Sourceval pack_arr : arr -> t
Sourceval unpack_arr : t -> arr
Sourceval pack_flt : float -> t
Sourceval unpack_flt : t -> float
Sourceval deep_info : t -> string
Sourceval type_info : t -> string
Sourceval error_binop : string -> t -> t -> 'a
Sourceval error_uniop : string -> t -> 'a
Sourcemodule Maths : sig ... end
Sourceval reverse_reset : t -> unit
Sourceval reverse_push : t -> t -> unit
Sourceval reverse_prop : t -> t -> unit
Sourceval make_forward : t -> t -> int -> t
Sourceval make_reverse : t -> int -> t
Sourceval diff' : (t -> t) -> t -> t * t
Sourceval diff : (t -> t) -> t -> t
Sourceval grad' : (t -> t) -> t -> t * t
Sourceval grad : (t -> t) -> t -> t
Sourceval jacobianv' : (t -> t) -> t -> t -> t * t
Sourceval jacobianv : (t -> t) -> t -> t -> t
Sourceval jacobianTv' : (t -> t) -> t -> t -> t * t
Sourceval jacobianTv : (t -> t) -> t -> t -> t
Sourceval jacobian' : (t -> t) -> t -> t * t
Sourceval jacobian : (t -> t) -> t -> t
Sourceval gradhessian : (t -> t) -> t -> t * t
Sourceval gradhessian' : (t -> t) -> t -> t * t * t
Sourceval hessian : (t -> t) -> t -> t
Sourceval hessian' : (t -> t) -> t -> t * t
Sourceval gradhessianv' : (t -> t) -> t -> t -> t * t * t
Sourceval gradhessianv : (t -> t) -> t -> t -> t * t
Sourceval hessianv' : (t -> t) -> t -> t -> t * t
Sourceval hessianv : (t -> t) -> t -> t -> t
Sourceval laplacian : (t -> t) -> t -> t
Sourceval laplacian' : (t -> t) -> t -> t * t
Sourcemodule Mat : sig ... end
Sourcemodule Arr : sig ... end
Sourceval _traverse_trace : t list -> (t, int * string * t list) Hashtbl.t
Sourceval _convert_terminal_output : (t, int * string * t list) Hashtbl.t -> string
Sourceval _convert_dot_output : (t, int * string * t list) Hashtbl.t -> string
Sourceval to_trace : t list -> string
Sourceval to_dot : t list -> string
Sourceval pp_num : Format.formatter -> t -> unit
Sourcemodule Utils : sig ... end
Sourcemodule Learning_Rate : sig ... end
Sourcemodule Batch : sig ... end
Sourcemodule Loss : sig ... end
Sourcemodule Gradient : sig ... end
Sourcemodule Momentum : sig ... end
Sourcemodule Regularisation : sig ... end
Sourcemodule Clipping : sig ... end
Sourcemodule Stopping : sig ... end
Sourcemodule Checkpoint : sig ... end
Sourcemodule Params : sig ... end
Sourceval minimise_weight : ?state:Checkpoint.state -> Params.typ -> (t -> t -> t) -> t -> t -> t -> Checkpoint.state * t
Sourceval 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
Sourceval minimise_fun : ?state:Checkpoint.state -> Params.typ -> (t -> t) -> t -> Checkpoint.state * t