Module Owl_neural_neuron

module Owl_neural_neuron: sig .. end

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