package owl

  1. Overview
  2. Docs
Legend:
Library
Module
Module type
Parameter
Class
Class type
type typ =
  1. | Relu
  2. | Sigmoid
  3. | Softmax
  4. | Tanh
  5. | Custom of t -> t
  6. | None
type neuron_typ = {
  1. mutable activation : typ;
  2. mutable in_shape : int array;
  3. mutable out_shape : int array;
}
val create : typ -> neuron_typ
val connect : int array -> neuron_typ -> unit
val run_activation : Owl_algodiff.S.t -> typ -> Owl_algodiff.S.t
val activation_to_string : typ -> string
val to_string : neuron_typ -> string