pub struct ReformerQuestionAnsweringModelOutput {
pub start_logits: Tensor,
pub end_logits: Tensor,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Tensor>>,
}Expand description
Container holding a Reformer model with question answering head
Fields§
§start_logits: Tensorstart logits
end_logits: Tensorend logits
Hidden states for all intermediate layers
all_attentions: Option<Vec<Tensor>>Attention weights for all intermediate layers
Auto Trait Implementations§
impl RefUnwindSafe for ReformerQuestionAnsweringModelOutput
impl Send for ReformerQuestionAnsweringModelOutput
impl !Sync for ReformerQuestionAnsweringModelOutput
impl Unpin for ReformerQuestionAnsweringModelOutput
impl UnwindSafe for ReformerQuestionAnsweringModelOutput
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