package torch

  1. Overview
  2. Docs
type t
type state = Tensor.t * Tensor.t
val create : Var_store.t -> input_dim:Base.int -> hidden_size:Base.int -> t
val step : t -> state -> Tensor.t -> state
val seq : t -> Tensor.t -> Tensor.t * state
val zero_state : t -> batch_size:Base.int -> state
OCaml

Innovation. Community. Security.