tensorflow

TensorFlow bindings for OCaml
IN THIS PACKAGE
type t = {
f : 'a. self:[< `float | `double ] as 'a Node.t -> gradient:'a Node.t -> Node.p Base.option Base.list;
}
val add : Tensorflow.Node.Op_name.t -> t -> Base.unit
val find : Tensorflow.Node.Op_name.t -> ( self:Node.p -> gradient:Node.p -> Node.p Base.option Base.list ) Base.option
type multi = {
g : 'a. self:[< `float | `double ] as 'a Node.t -> gradients:'a Node.t Base.Map.M(Base.Int).t -> Node.p Base.option Base.list;
}
val add_multi : Tensorflow.Node.Op_name.t -> multi -> Base.unit
val find_multi : Tensorflow.Node.Op_name.t -> ( self:Node.p -> gradients:Node.p Base.Map.M(Base.Int).t -> Node.p Base.option Base.list ) Base.option