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