package owl

  1. Overview
  2. Docs
Legend:
Library
Module
Module type
Parameter
Class
Class type
module Init : sig ... end
module Input : sig ... end
module Activation : sig ... end
module Linear : sig ... end
module LinearNoBias : sig ... end
module Recurrent : sig ... end
module LSTM : sig ... end
module GRU : sig ... end
module Conv1D : sig ... end
module Conv2D : sig ... end
module Conv3D : sig ... end
module FullyConnected : sig ... end
module MaxPool1D : sig ... end
module AvgPool1D : sig ... end
module MaxPool2D : sig ... end
module AvgPool2D : sig ... end
module UpSampling1D : sig ... end
module UpSampling2D : sig ... end
module UpSampling3D : sig ... end
module Padding1D : sig ... end
module Padding2D : sig ... end
module Padding3D : sig ... end
module Lambda : sig ... end
module Dropout : sig ... end
module Reshape : sig ... end
module Flatten : sig ... end
module Add : sig ... end
module Mul : sig ... end
module Dot : sig ... end
module Max : sig ... end
module Average : sig ... end
module Concatenate : sig ... end
module Normalisation : sig ... end
module GaussianNoise : sig ... end
module GaussianDropout : sig ... end
module Masking : sig ... end
type neuron =
  1. | Input of Input.neuron_typ
  2. | Linear of Linear.neuron_typ
  3. | LinearNoBias of LinearNoBias.neuron_typ
  4. | LSTM of LSTM.neuron_typ
  5. | GRU of GRU.neuron_typ
  6. | Recurrent of Recurrent.neuron_typ
  7. | Conv2D of Conv2D.neuron_typ
  8. | Conv3D of Conv3D.neuron_typ
  9. | FullyConnected of FullyConnected.neuron_typ
  10. | MaxPool2D of MaxPool2D.neuron_typ
  11. | AvgPool2D of AvgPool2D.neuron_typ
  12. | Dropout of Dropout.neuron_typ
  13. | Reshape of Reshape.neuron_typ
  14. | Flatten of Flatten.neuron_typ
  15. | Lambda of Lambda.neuron_typ
  16. | Activation of Activation.neuron_typ
  17. | Add of Add.neuron_typ
  18. | Mul of Mul.neuron_typ
  19. | Dot of Dot.neuron_typ
  20. | Max of Max.neuron_typ
  21. | Average of Average.neuron_typ
val get_in_out_shape : neuron -> int array * int array
val get_in_shape : neuron -> int array
val get_out_shape : neuron -> int array
val connect : int array -> neuron -> unit
val init : neuron -> unit
val reset : neuron -> unit
val mktag : int -> neuron -> unit
val mkpar : neuron -> t array
val mkpri : neuron -> Owl_algodiff.S.t array
val mkadj : neuron -> Owl_algodiff.S.t array
val update : neuron -> Owl_algodiff.S.t array -> unit
val run_array : Owl_algodiff.S.t array -> neuron -> Owl_algodiff.S.t
val to_string : neuron -> string