pub struct ReformerQuestionAnsweringModelOutput {
pub start_logits: Tensor,
pub end_logits: Tensor,
pub all_hidden_states: Option<Vec<Tensor>>,
pub all_attentions: Option<Vec<Tensor>>,
}Expand description
Container holding a Reformer model with question answering head
Fields§
§start_logits: Tensorstart logits
end_logits: Tensorend logits
Hidden states for all intermediate layers
all_attentions: Option<Vec<Tensor>>Attention weights for all intermediate layers