Struct rust_bert::reformer::ReformerClassificationOutput [−][src]
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: Tensorlogits
Hidden states for all intermediate layers
all_attentions: Option<Vec<Tensor>>Attention weights for all intermediate layers
Auto Trait Implementations
impl Send for ReformerClassificationOutput
impl !Sync for ReformerClassificationOutput
impl Unpin for ReformerClassificationOutput
impl UnwindSafe for ReformerClassificationOutput
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
