Make (p.owl-base.1.2.doc.owl-base.Owl_neural

Parameters

module Optimise : Owl_optimise_generic_sig.Sig

Signature

Sourcemodule Optimise = Optimise

Sourcemodule Init : sig ... end

Sourcemodule Input : sig ... end

Sourcemodule Activation : sig ... end

Sourcemodule Linear : sig ... end

Sourcemodule LinearNoBias : sig ... end

Sourcemodule Recurrent : sig ... end

Sourcemodule LSTM : sig ... end

Sourcemodule GRU : sig ... end

Sourcemodule Conv1D : sig ... end

Sourcemodule DilatedConv1D : sig ... end

Sourcemodule TransposeConv1D : sig ... end

Sourcemodule Conv2D : sig ... end

Sourcemodule DilatedConv2D : sig ... end

Sourcemodule TransposeConv2D : sig ... end

Sourcemodule Conv3D : sig ... end

Sourcemodule DilatedConv3D : sig ... end

Sourcemodule TransposeConv3D : sig ... end

Sourcemodule FullyConnected : sig ... end

Sourcemodule MaxPool1D : sig ... end

Sourcemodule MaxPool2D : sig ... end

Sourcemodule AvgPool1D : sig ... end

Sourcemodule AvgPool2D : sig ... end

Sourcemodule GlobalMaxPool1D : sig ... end

Sourcemodule GlobalMaxPool2D : sig ... end

Sourcemodule GlobalAvgPool1D : sig ... end

Sourcemodule GlobalAvgPool2D : sig ... end

Sourcemodule UpSampling1D : sig ... end

Sourcemodule UpSampling2D : sig ... end

Sourcemodule UpSampling3D : sig ... end

Sourcemodule Padding1D : sig ... end

Sourcemodule Padding2D : sig ... end

Sourcemodule Padding3D : sig ... end

Sourcemodule Lambda : sig ... end

Sourcemodule LambdaArray : sig ... end

Sourcemodule Dropout : sig ... end

Sourcemodule Reshape : sig ... end

Sourcemodule Flatten : sig ... end

Sourcemodule Slice : sig ... end

Sourcemodule Add : sig ... end

Sourcemodule Mul : sig ... end

Sourcemodule Dot : sig ... end

Sourcemodule Max : sig ... end

Sourcemodule Average : sig ... end

Sourcemodule Concatenate : sig ... end

Sourcemodule Normalisation : sig ... end

Sourcemodule GaussianNoise : sig ... end

Sourcemodule GaussianDropout : sig ... end

Sourcemodule AlphaDropout : sig ... end

Sourcemodule Embedding : sig ... end

Sourcemodule Masking : sig ... end

Sourcetype neuron =

| Input of Input.neuron_typ
| Linear of Linear.neuron_typ
| LinearNoBias of LinearNoBias.neuron_typ
| Embedding of Embedding.neuron_typ
| LSTM of LSTM.neuron_typ
| GRU of GRU.neuron_typ
| Recurrent of Recurrent.neuron_typ
| Conv1D of Conv1D.neuron_typ
| Conv2D of Conv2D.neuron_typ
| Conv3D of Conv3D.neuron_typ
| DilatedConv1D of DilatedConv1D.neuron_typ
| DilatedConv2D of DilatedConv2D.neuron_typ
| DilatedConv3D of DilatedConv3D.neuron_typ
| TransposeConv1D of TransposeConv1D.neuron_typ
| TransposeConv2D of TransposeConv2D.neuron_typ
| TransposeConv3D of TransposeConv3D.neuron_typ
| FullyConnected of FullyConnected.neuron_typ
| MaxPool1D of MaxPool1D.neuron_typ
| MaxPool2D of MaxPool2D.neuron_typ
| AvgPool1D of AvgPool1D.neuron_typ
| AvgPool2D of AvgPool2D.neuron_typ
| GlobalMaxPool1D of GlobalMaxPool1D.neuron_typ
| GlobalMaxPool2D of GlobalMaxPool2D.neuron_typ
| GlobalAvgPool1D of GlobalAvgPool1D.neuron_typ
| GlobalAvgPool2D of GlobalAvgPool2D.neuron_typ
| UpSampling2D of UpSampling2D.neuron_typ
| Padding2D of Padding2D.neuron_typ
| Dropout of Dropout.neuron_typ
| Reshape of Reshape.neuron_typ
| Flatten of Flatten.neuron_typ
| Slice of Slice.neuron_typ
| Lambda of Lambda.neuron_typ
| LambdaArray of LambdaArray.neuron_typ
| Activation of Activation.neuron_typ
| GaussianNoise of GaussianNoise.neuron_typ
| GaussianDropout of GaussianDropout.neuron_typ
| AlphaDropout of AlphaDropout.neuron_typ
| Normalisation of Normalisation.neuron_typ
| Add of Add.neuron_typ
| Mul of Mul.neuron_typ
| Dot of Dot.neuron_typ
| Max of Max.neuron_typ
| Average of Average.neuron_typ
| Concatenate of Concatenate.neuron_typ

Sourceval get_in_out_shape : neuron -> int array * int array

Sourceval get_in_shape : neuron -> int array

Sourceval get_out_shape : neuron -> int array

Sourceval connect : int array array -> neuron -> unit

Sourceval init : neuron -> unit

Sourceval reset : neuron -> unit

Sourceval mktag : int -> neuron -> unit

Sourceval mkpar : neuron -> Optimise.Algodiff.t array

Sourceval mkpri : neuron -> Optimise.Algodiff.t array

Sourceval mkadj : neuron -> Optimise.Algodiff.t array

Sourceval update : neuron -> Optimise.Algodiff.t array -> unit

Sourceval save_weights : neuron -> Optimise.Algodiff.t array

Sourceval load_weights : neuron -> Optimise.Algodiff.t array -> unit

Sourceval copy : neuron -> neuron

Sourceval run : Optimise.Algodiff.t array -> neuron -> Optimise.Algodiff.t

Sourceval to_string : neuron -> string

Sourceval to_name : neuron -> string