Struct rust_bert::bert::BertLayerOutput [−][src]
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 states
attention_weights: Option<Tensor>Self attention scores
cross_attention_weights: Option<Tensor>Cross attention scores
Auto Trait Implementations
impl RefUnwindSafe for BertLayerOutput
impl Send for BertLayerOutput
impl !Sync for BertLayerOutput
impl Unpin for BertLayerOutput
impl UnwindSafe for BertLayerOutput
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
