Struct InnerCarry 

pub struct InnerCarry {
    pub z_h: Tensor,
    pub z_l: Tensor,
}

State carry for recursive computation

Holds the high-level and low-level states that are refined across recursive cycles.

Fields

z_h: Tensor

High-level state: [batch, seq_len, hidden_size]

z_l: Tensor

Low-level state: [batch, seq_len, hidden_size]

Implementations

impl InnerCarry

pub fn new(z_h: Tensor, z_l: Tensor) -> Self

Create new carry with given states

pub fn empty( batch_size: usize, seq_len: usize, hidden_size: usize, dtype: DType, device: &Device, ) -> Result<Self>

Create empty carry (uninitialized tensors)

Trait Implementations

impl Clone for InnerCarry

fn clone(&self) -> InnerCarry

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for InnerCarry

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

Formats the value using the given formatter.