Struct rust_bert::models::bert::BertEncoderOutput
source · pub struct BertEncoderOutput {
pub hidden_state: Tensor,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Tensor>>,
}
Expand description
Container for the BERT encoder output.
Fields§
Last hidden states from the model
Hidden states for all intermediate layers
all_attentions: Option<Vec<Tensor>>
Attention weights for all intermediate layers
Auto Trait Implementations§
impl RefUnwindSafe for BertEncoderOutput
impl Send for BertEncoderOutput
impl !Sync for BertEncoderOutput
impl Unpin for BertEncoderOutput
impl UnwindSafe for BertEncoderOutput
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more