Function candle_nn::rnn::lstm

source · 
pub fn lstm(
    in_dim: usize,
    hidden_dim: usize,
    config: LSTMConfig,
    vb: VarBuilder<'_>
) -> Result<LSTM>
Expand description

Creates a LSTM layer.