[−][src]Struct rust_bert::bart::LayerState

pub struct LayerState {
    pub prev_key: Option<Tensor>,
    pub prev_value: Option<Tensor>,
    pub prev_key_padding_mask: Option<Tensor>,
}

Cache for BART attention layers

Stores the cached value of key, value and key padding mask to avoid recalculation (e.g. at each generation step)

Fields

prev_key: Option<Tensor>

Cached keys

prev_value: Option<Tensor>

Cached values

prev_key_padding_mask: Option<Tensor>

Cached keys padding mask

Trait Implementations

impl Debug for LayerState[src]

Auto Trait Implementations

Blanket Implementations

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

impl<T> Borrow<T> for T where
    T: ?Sized, [src]

impl<T> BorrowMut<T> for T where
    T: ?Sized, [src]

impl<T> From<T> for T[src]

impl<T, U> Into<U> for T where
    U: From<T>, [src]

impl<T, U> TryFrom<U> for T where
    U: Into<T>, [src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, [src]

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

The type returned in the event of a conversion error.