Struct ModelOutput

pub struct ModelOutput {
    pub logits: Tensor,
    pub hidden_states: Option<Vec<Tensor>>,
    pub attentions: Option<Vec<Tensor>>,
}

Amplify Model Output

logits, hidden states, and attentions.

logits -> distribution of the sequences. attentions -> contact map

Fields

logits: Tensor

hidden_states: Option<Vec<Tensor>>

attentions: Option<Vec<Tensor>>

Implementations

impl ModelOutput

pub fn get_contact_map(&self) -> Result<Option<Tensor>>

Contact maps can be obtained from the self-attentions

Trait Implementations

impl Debug for ModelOutput

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

Formats the value using the given formatter.