owl 0.2.6 · OCaml Package

owl

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 =

| Input of Input.neuron_typ
| Linear of Linear.neuron_typ
| LinearNoBias of LinearNoBias.neuron_typ
| LSTM of LSTM.neuron_typ
| GRU of GRU.neuron_typ
| Recurrent of Recurrent.neuron_typ
| Conv2D of Conv2D.neuron_typ
| Conv3D of Conv3D.neuron_typ
| FullyConnected of FullyConnected.neuron_typ
| MaxPool2D of MaxPool2D.neuron_typ
| AvgPool2D of AvgPool2D.neuron_typ
| Dropout of Dropout.neuron_typ
| Reshape of Reshape.neuron_typ
| Flatten of Flatten.neuron_typ
| Lambda of Lambda.neuron_typ
| Activation of Activation.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

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

package owl