Module candle_nn::rnn

Recurrent Neural Networks

Structs

• GRU
  A Gated Recurrent Unit (GRU) layer.
• GRUConfig
• GRUState
  The state for a GRU network, this contains a single tensor.
• LSTM
  A Long Short-Term Memory (LSTM) layer.
• LSTMConfig
• LSTMState
  The state for a LSTM network, this contains two tensors.

Traits

• RNN
  Trait for Recurrent Neural Networks.

Functions

• gru
  Creates a GRU layer.
• lstm
  Creates a LSTM layer.