Struct rust_bert::models::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
Auto Trait Implementations§
impl RefUnwindSafe for DebertaMaskedLMOutput
impl Send for DebertaMaskedLMOutput
impl !Sync for DebertaMaskedLMOutput
impl Unpin for DebertaMaskedLMOutput
impl UnwindSafe for DebertaMaskedLMOutput
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more