Function gru

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

Creates a GRU layer.