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