Struct rust_bert::reformer::LayerState

pub struct LayerState {
    pub prev_buckets: Option<Tensor>,
    pub prev_states: Tensor,
}

Cache for Reformer attention layers

Stores the cached value of buckets and states to avoid recalculation (e.g. at each generation step)

Fields

prev_buckets: Option<Tensor>

Cached buckets

prev_states: Tensor

Cached states

Trait Implementations

impl Clone for LayerState

fn clone(&self) -> Self

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for LayerState

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

Formats the value using the given formatter.