Struct rust_bert::longformer::LongformerTokenClassificationOutput [−][src]
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>>,
}Expand description
Container for the Longformer token classification model output.
Fields
logits: TensorLogits for each sequence item (token) for each target class
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
Auto Trait Implementations
impl Send for LongformerTokenClassificationOutput
impl !Sync for LongformerTokenClassificationOutput
impl Unpin for LongformerTokenClassificationOutput
Blanket Implementations
Mutably borrows from an owned value. Read more
Instruments this type with the provided Span, returning an
Instrumented wrapper. Read more
type Output = T
type Output = T
Should always be Self
