pub struct DistilBertQuestionAnsweringOutput {
pub start_logits: Tensor,
pub end_logits: Tensor,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Tensor>>,
}Expand description
Container for the DistilBERT question answering model output
Fields§
§start_logits: TensorLogits for the start position for token of each input sequence
end_logits: TensorLogits for the end position for token of each input sequence
Hidden states for all intermediate layers
all_attentions: Option<Vec<Tensor>>Attention weights for all intermediate layers
Auto Trait Implementations§
impl RefUnwindSafe for DistilBertQuestionAnsweringOutput
impl Send for DistilBertQuestionAnsweringOutput
impl !Sync for DistilBertQuestionAnsweringOutput
impl Unpin for DistilBertQuestionAnsweringOutput
impl UnwindSafe for DistilBertQuestionAnsweringOutput
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