Struct rust_bert::albert::AlbertSequenceClassificationOutput

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

Container for the ALBERT sequence classification model

Fields

logits: Tensor

Logits for each input (sequence) for each target class

all_hidden_states: Option<Vec<Tensor>>

Hidden states for all intermediate layers

all_attentions: Option<Vec<Vec<Tensor>>>

Attention weights for all intermediate layers