Struct rust_bert::distilbert::DistilBertMaskedLMOutput

pub struct DistilBertMaskedLMOutput {
    pub prediction_scores: Tensor,
    pub all_hidden_states: Option<Vec<Tensor>>,
    pub all_attentions: Option<Vec<Tensor>>,
}

Container for the DistilBERT masked LM model output.

Fields

prediction_scores: Tensor

Logits for the vocabulary items at each sequence position

all_hidden_states: Option<Vec<Tensor>>

Hidden states for all intermediate layers

all_attentions: Option<Vec<Tensor>>

Attention weights for all intermediate layers