Previous  Up  Next

Module Owl_neural.Activation

module Activation: Owl_neural_neuron.Activation
type typ =
| Relu
| Sigmoid
| Softmax
| Tanh
| Custom of (Owl_neural_neuron.t -> Owl_neural_neuron.t)
| None 
type neuron_typ = {
   mutable activation : typ;
   mutable in_shape : int array;
   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 run : Owl_algodiff.S.t ->
       neuron_typ -> Owl_algodiff.S.t
val activation_to_string : typ -> string
val to_string : neuron_typ -> string