Struct rust_bert::deberta::DebertaMaskedLMOutput
source · pub struct DebertaMaskedLMOutput {
pub logits: Tensor,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Tensor>>,
}Expand description
Container for the DeBERTa masked LM model output.
Fields§
§logits: 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