Struct GRU 

pub struct GRU<B>
where
    B: Backend,
{ /* private fields */ }

A multi-step GRU that unrolls a GRUCell over the sequence dimension.

Shapes

• input: [batch, seq_len, input_size]
• output: [batch, seq_len, hidden_size]
• h_n: [batch, hidden_size]

Implementations

impl<B> GRU<B>

where B: Backend,

pub fn new( input_size: usize, hidden_size: usize, use_bias: bool, dtype: DType, device: &<B as Backend>::Device, ) -> Result<GRU<B>, Error>

Create a new GRU layer.

pub fn forward( &self, x: &Tensor<B>, h0: Option<&Tensor<B>>, ) -> Result<(Tensor<B>, Tensor<B>), Error>

Forward pass over the full sequence.

• x: [batch, seq_len, input_size]
• h0: optional initial hidden state [batch, hidden_size]. If None, zeros are used.

Returns (output, h_n) where:

• output: [batch, seq_len, hidden_size]
• h_n: [batch, hidden_size]

pub fn parameters(&self) -> Vec<Tensor<B>>

Return all trainable parameters.

pub fn named_parameters(&self) -> Vec<(String, Tensor<B>)>

Return all trainable parameters with names.

pub fn cell(&self) -> &GRUCell<B>

Access the underlying cell.