Struct rust_bert::electra::ElectraDiscriminatorOutput

pub struct ElectraDiscriminatorOutput {
    pub probabilities: Tensor,
    pub all_hidden_states: Option<Vec<Tensor>>,
    pub all_attentions: Option<Vec<Tensor>>,
}

Container for the Electra discriminator model output.

Fields

probabilities: Tensor

Probabilities for each sequence item (token) to be generated by a language model

all_hidden_states: Option<Vec<Tensor>>

Hidden states for all intermediate layers

all_attentions: Option<Vec<Tensor>>

Attention weights for all intermediate layers