Type Alias DebertaQuestionAnsweringOutput

pub type DebertaQuestionAnsweringOutput = BertQuestionAnsweringOutput;

Container for the DeBERTa question answering model output.

Aliased Type

struct DebertaQuestionAnsweringOutput {
    pub start_logits: Tensor,
    pub end_logits: Tensor,
    pub all_hidden_states: Option<Vec<Tensor>>,
    pub all_attentions: Option<Vec<Tensor>>,
}

Fields

start_logits: Tensor

Logits for the start position for token of each input sequence

end_logits: Tensor

Logits for the end position for token of each input sequence

all_hidden_states: Option<Vec<Tensor>>

Hidden states for all intermediate layers

all_attentions: Option<Vec<Tensor>>

Attention weights for all intermediate layers