Struct rust_bert::bert::BertModelOutput [−][src]
pub struct BertModelOutput {
pub hidden_state: Tensor,
pub pooled_output: Option<Tensor>,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Tensor>>,
}Expand description
Container for the BERT model output.
Fields
Last hidden states from the model
pooled_output: Option<Tensor>Pooled output (hidden state for the first token)
Hidden states for all intermediate layers
all_attentions: Option<Vec<Tensor>>Attention weights for all intermediate layers
Auto Trait Implementations
impl RefUnwindSafe for BertModelOutput
impl Send for BertModelOutput
impl !Sync for BertModelOutput
impl Unpin for BertModelOutput
impl UnwindSafe for BertModelOutput
Blanket Implementations
Mutably borrows from an owned value. Read more
Instruments this type with the provided Span, returning an
Instrumented wrapper. Read more
type Output = T
type Output = T
Should always be Self
