Function tch::nn::lstm

pub fn lstm<'a, T: Borrow<Path<'a>>>(
    vs: T,
    in_dim: i64,
    hidden_dim: i64,
    c: RNNConfig
) -> LSTM

Creates a LSTM layer.