package tensorflow

  1. Overview
  2. Docs
val gru : shape:int -> (h:(Nn._1d, [ `float ]) Nn.t -> x:(Nn._1d, [ `float ]) Nn.t -> (Nn._1d, [ `float ]) Nn.t) Core_kernel.Std.Staged.t
val fold : (Nn._2d, [ `float ]) Nn.t -> init:'a -> f:('a -> (Nn._1d, [ `float ]) Nn.t -> 'a) -> 'a
val scan : (Nn._2d, [ `float ]) Nn.t -> init:(Nn._1d, [ `float ]) Nn.t -> f: ((Nn._1d, [ `float ]) Nn.t -> (Nn._1d, [ `float ]) Nn.t -> (Nn._1d, [ `float ]) Nn.t) -> (Nn._2d, [ `float ]) Nn.t * (Nn._1d, [ `float ]) Nn.t
val scan' : (Nn._2d, [ `float ]) Nn.t -> init:'a -> f:('a -> (Nn._1d, [ `float ]) Nn.t -> (Nn._1d, [ `float ]) Nn.t * 'a) -> (Nn._2d, [ `float ]) Nn.t * 'a
OCaml

Innovation. Community. Security.