Struct rust_bert::models::albert::AlbertOutput
source · pub struct AlbertOutput {
pub hidden_state: Tensor,
pub pooled_output: Tensor,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Vec<Tensor>>>,
}
Expand description
Container for the ALBERT model output.
Fields§
Last hidden states from the model
pooled_output: Tensor
Pooled output (hidden state for the first token)
Hidden states for all intermediate layers
all_attentions: Option<Vec<Vec<Tensor>>>
Attention weights for all intermediate layers
Auto Trait Implementations§
impl RefUnwindSafe for AlbertOutput
impl Send for AlbertOutput
impl !Sync for AlbertOutput
impl Unpin for AlbertOutput
impl UnwindSafe for AlbertOutput
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