Struct rust_bert::bert::BertLayerOutput

source · 
pub struct BertLayerOutput {
    pub hidden_state: Tensor,
    pub attention_weights: Option<Tensor>,
    pub cross_attention_weights: Option<Tensor>,
}
Expand description

Container for the BERT layer output.

Fields§

§hidden_state: Tensor

Hidden states

§attention_weights: Option<Tensor>

Self attention scores

§cross_attention_weights: Option<Tensor>

Cross attention scores

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