Function candle_nn::rnn::gru

source ·
pub fn gru(
    in_dim: usize,
    hidden_dim: usize,
    config: GRUConfig,
    vb: VarBuilder<'_>
) -> Result<GRU>
Expand description

Creates a GRU layer.