Struct rust_bert::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§

§hidden_state: Tensor

Last hidden states from the model

§pooled_output: Tensor

Pooled output (hidden state for the first token)

§all_hidden_states: Option<Vec<Tensor>>

Hidden states for all intermediate layers

§all_attentions: Option<Vec<Vec<Tensor>>>

Attention weights for all intermediate layers

Auto Trait Implementations§

Blanket Implementations§

Gets the TypeId of self. Read more
Immutably borrows from an owned value. Read more
Mutably borrows from an owned value. Read more

Returns the argument unchanged.

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more
Instruments this type with the current Span, returning an Instrumented wrapper. Read more

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

The alignment of pointer.
The type for initializers.
Initializes a with the given initializer. Read more
Dereferences the given pointer. Read more
Mutably dereferences the given pointer. Read more
Drops the object pointed to by the given pointer. Read more
Should always be Self
The type returned in the event of a conversion error.
Performs the conversion.
The type returned in the event of a conversion error.
Performs the conversion.
Attaches the provided Subscriber to this type, returning a WithDispatch wrapper. Read more
Attaches the current default Subscriber to this type, returning a WithDispatch wrapper. Read more