diffusion_rs_common::nn::rnn

Function lstm

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

Creates a LSTM layer.