Function tch::nn::lstm

pub fn lstm(vs: &Path<'_>, in_dim: i64, hidden_dim: i64, c: RNNConfig) -> LSTM

Creates a LSTM layer.