Struct rust_bert::distilbert::DistilBertMaskedLMOutput [−][src]
pub struct DistilBertMaskedLMOutput {
pub prediction_scores: Tensor,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Tensor>>,
}Expand description
Container for the DistilBERT masked LM model output.
Fields
prediction_scores: TensorLogits for the vocabulary items at each sequence position
Hidden states for all intermediate layers
all_attentions: Option<Vec<Tensor>>Attention weights for all intermediate layers
Auto Trait Implementations
impl RefUnwindSafe for DistilBertMaskedLMOutput
impl Send for DistilBertMaskedLMOutput
impl !Sync for DistilBertMaskedLMOutput
impl Unpin for DistilBertMaskedLMOutput
impl UnwindSafe for DistilBertMaskedLMOutput
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
