Neuron.Lambda
type neuron_typ =
Owl_neural_generic.Make_Embedded(Engine).Neuron.Lambda.neuron_typ =
{
mutable lambda : Optimise.Algodiff.t -> Optimise.Algodiff.t;
mutable in_shape : int array;
mutable out_shape : int array;
}
val create :
?out_shape:int array ->
(Optimise.Algodiff.t -> Optimise.Algodiff.t) ->
neuron_typ
val connect : int array -> neuron_typ -> unit
val copy : neuron_typ -> neuron_typ
val run : Optimise.Algodiff.t -> neuron_typ -> Optimise.Algodiff.t
val to_string : neuron_typ -> string