Function tch::nn::gru

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

Creates a new GRU layer.