Struct GRU

pub struct GRU { /* private fields */ }

A Gated Recurrent Unit (GRU) layer.

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

Implementations

impl GRU

pub fn new( in_dim: usize, hidden_dim: usize, config: GRUConfig, vb: VarBuilder<'_>, ) -> Result<Self>

Creates a GRU layer.

pub fn config(&self) -> &GRUConfig

Trait Implementations

impl Clone for GRU

fn clone(&self) -> GRU

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

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: usize) -> Result<Self::State>

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

fn step(&self, input: &Tensor, in_state: &Self::State) -> Result<Self::State>

Applies a single step of the recurrent network.

fn states_to_tensor(&self, states: &[Self::State]) -> Result<Tensor>

Converts a sequence of state to a tensor.

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

Applies multiple steps of the recurrent network.

fn seq_init( &self, input: &Tensor, init_state: &Self::State, ) -> Result<Vec<Self::State>>

Applies multiple steps of the recurrent network.