Struct rust_bert::models::reformer::ReformerClassificationOutput
source · pub struct ReformerClassificationOutput {
pub logits: Tensor,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Tensor>>,
}
Expand description
Container holding a Reformer model with classification head
Fields§
§logits: Tensor
logits
Hidden states for all intermediate layers
all_attentions: Option<Vec<Tensor>>
Attention weights for all intermediate layers
Auto Trait Implementations§
impl RefUnwindSafe for ReformerClassificationOutput
impl Send for ReformerClassificationOutput
impl !Sync for ReformerClassificationOutput
impl Unpin for ReformerClassificationOutput
impl UnwindSafe for ReformerClassificationOutput
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