GRUCell

rust_lstm::layers::gru_cell

Struct GRUCell 

Source 
pub struct GRUCell {
Show 17 fields
    pub w_ir: Array2<f64>,
    pub w_hr: Array2<f64>,
    pub b_ir: Array2<f64>,
    pub b_hr: Array2<f64>,
    pub w_iz: Array2<f64>,
    pub w_hz: Array2<f64>,
    pub b_iz: Array2<f64>,
    pub b_hz: Array2<f64>,
    pub w_ih: Array2<f64>,
    pub w_hh: Array2<f64>,
    pub b_ih: Array2<f64>,
    pub b_hh: Array2<f64>,
    pub hidden_size: usize,
    pub input_dropout: Option<Dropout>,
    pub recurrent_dropout: Option<Dropout>,
    pub output_dropout: Option<Dropout>,
    pub is_training: bool,

}
Expand description

GRU cell with trainable parameters and dropout support

Fields§

§w_ir: Array2<f64>§w_hr: Array2<f64>§b_ir: Array2<f64>§b_hr: Array2<f64>§w_iz: Array2<f64>§w_hz: Array2<f64>§b_iz: Array2<f64>§b_hz: Array2<f64>§w_ih: Array2<f64>§w_hh: Array2<f64>§b_ih: Array2<f64>§b_hh: Array2<f64>§hidden_size: usize§input_dropout: Option<Dropout>§recurrent_dropout: Option<Dropout>§output_dropout: Option<Dropout>§is_training: bool

Implementations§

Source§

impl GRUCell

Source

pub fn new(input_size: usize, hidden_size: usize) -> Self

Creates new GRU cell with Xavier-uniform weight initialization

Source

pub fn with_input_dropout(self, dropout_rate: f64, variational: bool) -> Self

Source

pub fn with_recurrent_dropout( self, dropout_rate: f64, variational: bool, ) -> Self

Source

pub fn with_output_dropout(self, dropout_rate: f64) -> Self

Source

pub fn train(&mut self)

Source

pub fn eval(&mut self)

Source

pub fn forward(&mut self, input: &Array2<f64>, hx: &Array2<f64>) -> Array2<f64>

Source

pub fn forward_with_cache( &mut self, input: &Array2<f64>, hx: &Array2<f64>, ) -> (Array2<f64>, GRUCellCache)

Source

pub fn backward( &self, dhy: &Array2<f64>, cache: &GRUCellCache, ) -> (GRUCellGradients, Array2<f64>, Array2<f64>)

Backward pass implementing GRU gradient computation with dropout

Returns (parameter_gradients, input_gradient, hidden_gradient)

Source

pub fn zero_gradients(&self) -> GRUCellGradients

Initialize zero gradients for accumulation

Source

pub fn update_parameters<O: Optimizer>( &mut self, gradients: &GRUCellGradients, optimizer: &mut O, prefix: &str, )

Apply gradients using the provided optimizer

Trait Implementations§

Source§

impl Clone for GRUCell

Source§

fn clone(&self) -> GRUCell

Returns a duplicate of the value. Read more
1.0.0 · Source§

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

Performs copy-assignment from source. Read more

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V