Struct tch::nn::GRU

pub struct GRU { /* fields omitted */ }

A Gated Recurrent Unit (GRU) layer.

https://en.wikipedia.org/wiki/Gated_recurrent_unit

Trait Implementations

impl Debug for GRU

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl RNN for GRU

type State = GRUState

fn zero_state(&self, batch_dim: i64) -> GRUState

A zero state from which the recurrent network is usually initialized.

fn step(&self, input: &Tensor, in_state: &GRUState) -> GRUState

Applies a single step of the recurrent network.

fn seq_init(&self, input: &Tensor, in_state: &GRUState) -> (Tensor, GRUState)

Applies multiple steps of the recurrent network.

fn seq(&self, input: &Tensor) -> (Tensor, Self::State)

Applies multiple steps of the recurrent network.