rust_bert::models::mbart

Type Alias LayerState

Source 
pub type LayerState = LayerState;

Aliased Type§

struct LayerState {
    pub prev_key: Tensor,
    pub prev_value: Tensor,
}

Fields§

§prev_key: Tensor

Cached keys

§prev_value: Tensor

Cached values