Function tch::nn::gru

source · []
pub fn gru<'a, T: Borrow<Path<'a>>>(
    vs: T,
    in_dim: i64,
    hidden_dim: i64,
    c: RNNConfig
) -> GRU
Expand description

Creates a new GRU layer.