tensorflow

TensorFlow bindings for OCaml
IN THIS PACKAGE
Module Tensorflow_fnn . Cell
val lstm : size_c:Base.int -> size_x:Base.int -> ( h:[ `float ] Tensorflow.Node.t -> x:[ `float ] Tensorflow.Node.t -> c:[ `float ] Tensorflow.Node.t -> [ `h of [ `float ] Tensorflow.Node.t ] * [ `c of [ `float ] Tensorflow.Node.t ] ) Base.Staged.t
val lstm_d : size_c:Base.int -> size_x:Base.int -> ( h:[ `double ] Tensorflow.Node.t -> x:[ `double ] Tensorflow.Node.t -> c:[ `double ] Tensorflow.Node.t -> [ `h of [ `double ] Tensorflow.Node.t ] * [ `c of [ `double ] Tensorflow.Node.t ] ) Base.Staged.t
val gru : size_h:Base.int -> size_x:Base.int -> ( h:[ `float ] Tensorflow.Node.t -> x:[ `float ] Tensorflow.Node.t -> [ `float ] Tensorflow.Node.t ) Base.Staged.t
val gru_d : size_h:Base.int -> size_x:Base.int -> ( h:[ `double ] Tensorflow.Node.t -> x:[ `double ] Tensorflow.Node.t -> [ `double ] Tensorflow.Node.t ) Base.Staged.t
module Unfold : sig ... end