Struct rust_bert::models::electra::ElectraDiscriminatorOutput
source · pub struct ElectraDiscriminatorOutput {
pub probabilities: Tensor,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Tensor>>,
}Expand description
Container for the Electra discriminator model output.
Fields§
§probabilities: TensorProbabilities for each sequence item (token) to be generated by a language model
Hidden states for all intermediate layers
all_attentions: Option<Vec<Tensor>>Attention weights for all intermediate layers
Auto Trait Implementations§
impl RefUnwindSafe for ElectraDiscriminatorOutput
impl Send for ElectraDiscriminatorOutput
impl !Sync for ElectraDiscriminatorOutput
impl Unpin for ElectraDiscriminatorOutput
impl UnwindSafe for ElectraDiscriminatorOutput
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