module Owl_neural_neuron: sig
.. end
Neural network: Neuron definitions
type
t = Owl_algodiff.S.t
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 =
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 : Owl_algodiff.S.t -> neuron -> Owl_algodiff.S.t
val run_array : Owl_algodiff.S.t array -> neuron -> Owl_algodiff.S.t
val to_string : neuron -> string