tensorflow

TensorFlow bindings for OCaml
IN THIS PACKAGE
Module Tensorflow . Layer . Linear
type 'a t
val create : int -> 'a t

create output_dim creates a linear layer with output size output_dim.

val apply : ?activation:activation -> ?use_bias:bool -> [< `double | `float ] as 'a t -> 'a Node.t -> 'a Node.t
val vars : 'a t -> 'a Node.t list