Struct rust_bert::models::longformer::LongformerTokenClassificationOutput

pub struct LongformerTokenClassificationOutput {
    pub logits: Tensor,
    pub all_hidden_states: Option<Vec<Tensor>>,
    pub all_attentions: Option<Vec<Tensor>>,
    pub all_global_attentions: Option<Vec<Tensor>>,
}

Container for the Longformer token classification model output.

Fields

logits: Tensor

Logits for each sequence item (token) for each target class

all_hidden_states: Option<Vec<Tensor>>

Hidden states for all intermediate layers

all_attentions: Option<Vec<Tensor>>

Attention weights for all intermediate layers

all_global_attentions: Option<Vec<Tensor>>

Global attention weights for all intermediate layers