package owl-base

  1. Overview
  2. Docs
package owl-base
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Make.Graph

module Neuron : sig ... end
Sourcetype node = Make_Embedded(A).node = {
  1. mutable name : string;
  2. mutable prev : node array;
  3. mutable next : node array;
  4. mutable neuron : Neuron.neuron;
  5. mutable output : Neuron.Optimise.Algodiff.t option;
  6. mutable network : network;
  7. mutable train : bool;
}
Sourceand network = Make_Embedded(A).network = {
  1. mutable nnid : string;
  2. mutable size : int;
  3. mutable roots : node array;
  4. mutable outputs : node array;
  5. mutable topo : node array;
}
Sourceval make_network : ?nnid:string -> int -> node array -> node array -> network
Sourceval make_node : ?name:string -> ?train:bool -> node array -> node array -> Neuron.neuron -> Neuron.Optimise.Algodiff.t option -> network -> node
Sourceval get_roots : network -> node array
Sourceval get_outputs : network -> node array
Sourceval get_node : network -> string -> node
Sourceval get_network : ?name:string -> node -> network
Sourceval outputs : ?name:string -> node array -> network
Sourceval get_network_name : network -> string
Sourceval set_network_name : network -> string -> unit
Sourceval collect_output : node array -> Neuron.Optimise.Algodiff.t array
Sourceval connect_pair : node -> node -> unit
Sourceval connect_to_parents : node array -> node -> unit
Sourceval add_node : ?act_typ:Neuron.Activation.typ -> network -> node array -> node -> node
Sourceval input_shape : network -> int array
Sourceval input_shapes : network -> int array array
Sourceval init : network -> unit
Sourceval reset : network -> unit
Sourceval mktag : int -> network -> unit
Sourceval mkpar : network -> Neuron.Optimise.Algodiff.t array array
Sourceval mkpri : network -> Neuron.Optimise.Algodiff.t array array
Sourceval mkadj : network -> Neuron.Optimise.Algodiff.t array array
Sourceval update : network -> Neuron.Optimise.Algodiff.t array array -> unit
Sourceval copy : network -> network
Sourceval input : ?name:string -> int array -> node
Sourceval inputs : ?names:string array -> int array array -> node array
Sourceval activation : ?name:string -> Neuron.Activation.typ -> node -> node
Sourceval linear : ?name:string -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int -> node -> node
Sourceval linear_nobias : ?name:string -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int -> node -> node
Sourceval embedding : ?name:string -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int -> int -> node -> node
Sourceval recurrent : ?name:string -> ?init_typ:Neuron.Init.typ -> act_typ:Neuron.Activation.typ -> int -> int -> node -> node
Sourceval lstm : ?name:string -> ?init_typ:Neuron.Init.typ -> int -> node -> node
Sourceval gru : ?name:string -> ?init_typ:Neuron.Init.typ -> int -> node -> node
Sourceval conv1d : ?name:string -> ?padding:Owl_types.padding -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval conv2d : ?name:string -> ?padding:Owl_types.padding -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval conv3d : ?name:string -> ?padding:Owl_types.padding -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval dilated_conv1d : ?name:string -> ?padding:Owl_types.padding -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> int array -> node -> node
Sourceval dilated_conv2d : ?name:string -> ?padding:Owl_types.padding -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> int array -> node -> node
Sourceval dilated_conv3d : ?name:string -> ?padding:Owl_types.padding -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> int array -> node -> node
Sourceval transpose_conv1d : ?name:string -> ?padding:Owl_types.padding -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval transpose_conv2d : ?name:string -> ?padding:Owl_types.padding -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval transpose_conv3d : ?name:string -> ?padding:Owl_types.padding -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval fully_connected : ?name:string -> ?init_typ:Neuron.Init.typ -> ?act_typ:Neuron.Activation.typ -> int -> node -> node
Sourceval max_pool1d : ?name:string -> ?padding:Owl_types.padding -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval max_pool2d : ?name:string -> ?padding:Owl_types.padding -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval avg_pool1d : ?name:string -> ?padding:Owl_types.padding -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval avg_pool2d : ?name:string -> ?padding:Owl_types.padding -> ?act_typ:Neuron.Activation.typ -> int array -> int array -> node -> node
Sourceval global_max_pool1d : ?name:string -> ?act_typ:Neuron.Activation.typ -> node -> node
Sourceval global_max_pool2d : ?name:string -> ?act_typ:Neuron.Activation.typ -> node -> node
Sourceval global_avg_pool1d : ?name:string -> ?act_typ:Neuron.Activation.typ -> node -> node
Sourceval global_avg_pool2d : ?name:string -> ?act_typ:Neuron.Activation.typ -> node -> node
Sourceval upsampling2d : ?name:string -> ?act_typ:Neuron.Activation.typ -> int array -> node -> node
Sourceval padding2d : ?name:string -> ?act_typ:Neuron.Activation.typ -> int array array -> node -> node
Sourceval dropout : ?name:string -> float -> node -> node
Sourceval gaussian_noise : ?name:string -> float -> node -> node
Sourceval gaussian_dropout : ?name:string -> float -> node -> node
Sourceval alpha_dropout : ?name:string -> float -> node -> node
Sourceval normalisation : ?name:string -> ?axis:int -> ?training:bool -> ?decay:float -> ?mu:Neuron.Optimise.Algodiff.A.arr -> ?var:Neuron.Optimise.Algodiff.A.arr -> node -> node
Sourceval reshape : ?name:string -> int array -> node -> node
Sourceval flatten : ?name:string -> node -> node
Sourceval lambda_array : ?name:string -> ?act_typ:Neuron.Activation.typ -> int array -> (Neuron.Optimise.Algodiff.t array -> Neuron.Optimise.Algodiff.t) -> node array -> node
Sourceval add : ?name:string -> ?act_typ:Neuron.Activation.typ -> node array -> node
Sourceval mul : ?name:string -> ?act_typ:Neuron.Activation.typ -> node array -> node
Sourceval dot : ?name:string -> ?act_typ:Neuron.Activation.typ -> node array -> node
Sourceval max : ?name:string -> ?act_typ:Neuron.Activation.typ -> node array -> node
Sourceval average : ?name:string -> ?act_typ:Neuron.Activation.typ -> node array -> node
Sourceval concatenate : ?name:string -> ?act_typ:Neuron.Activation.typ -> int -> node array -> node
Sourceval to_string : network -> string
Sourceval pp_network : Format.formatter -> network -> unit
Sourceval print : network -> unit
Sourceval save : ?unsafe:bool -> network -> string -> unit
Sourceval load : string -> network
Sourceval save_weights : network -> string -> unit
Sourceval load_weights : network -> string -> unit
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct