Struct ESM3Output 

pub struct ESM3Output {
    pub sequence_logits: Option<Tensor>,
    pub structure_logits: Option<Tensor>,
    pub secondary_structure_logits: Option<Tensor>,
    pub sasa_logits: Option<Tensor>,
    pub function_logits: Option<Tensor>,
    pub residue_logits: Option<Tensor>,
    pub embeddings: Option<Tensor>,
}

Output of an ESM3 forward pass.

Fields

sequence_logits: Option<Tensor>

(B, L, d_sequence_vocab) — per-residue amino-acid logits.

structure_logits: Option<Tensor>

(B, L, d_structure_vocab) — per-residue structure-token logits.

secondary_structure_logits: Option<Tensor>

(B, L, d_ss8_vocab) — secondary-structure logits.

sasa_logits: Option<Tensor>

(B, L, d_sasa_vocab) — SASA-bin logits.

function_logits: Option<Tensor>

(B, L, n_function_tracks, d_function_vocab) — function annotation logits.

residue_logits: Option<Tensor>

(B, L, d_residue_vocab) — InterPro residue-annotation logits.

embeddings: Option<Tensor>

(B, L, d_model) — final hidden states (when requested).

Trait Implementations

impl Debug for ESM3Output

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

Formats the value using the given formatter.