Function tch::nn::gru[][src]

pub fn gru(vs: &Path<'_>, in_dim: i64, hidden_dim: i64, c: RNNConfig) -> GRU
Expand description

Creates a new GRU layer.