Struct BiaffineScoreLogits 

pub struct BiaffineScoreLogits {
    pub head_score_logits: Tensor,
    pub relation_score_logits: Tensor,
    pub heads: Tensor,
}

Logits of a biaffine parsing layer.

Fields

head_score_logits: Tensor

Head score logits.

This tensor of shape [batch_size, seq_len, seq_len] contains for every sentence in the batch the scores for each head, given a dependency. For instance [s, d, h] is the score for dependent b being attached to head h in sentence s.

relation_score_logits: Tensor

This tensor of shape [batch_size, seq_len, seq_len, n_relations] contains dependency relation logits. For instance, if in sentence s, h is the head of d, then [s, d, h] is the a of the corresponding relation logits.

heads: Tensor

Tensor of shape [batch_size, seq_len] with heads found after MST decoding.

Trait Implementations

impl Debug for BiaffineScoreLogits

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.