package tensorflow

  1. Overview
  2. Docs
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
OCaml

Innovation. Community. Security.