Enum ModelOutputs 

pub enum ModelOutputs {
    Logits {
        logits: Tensor,
        hidden_states: Option<Tensor>,
    },
    Embeddings {
        embeddings: Tensor,
        pooled: Option<Tensor>,
    },
    Multimodal {
        text_logits: Option<Tensor>,
        image_logits: Option<Tensor>,
        text_embeddings: Option<Tensor>,
        image_embeddings: Option<Tensor>,
    },
    Sequence {
        logits: Tensor,
        encoder_hidden_states: Option<Tensor>,
        decoder_hidden_states: Option<Tensor>,
    },
    CLIP {
        logits_per_text: Tensor,
        logits_per_image: Tensor,
        text_embeds: Tensor,
        image_embeds: Tensor,
    },
}

Unified model outputs - supports all model types

Variants

Logits

Language model logits

Fields

logits: Tensor

hidden_states: Option<Tensor>

Embeddings

Embeddings/representations

Fields

embeddings: Tensor

pooled: Option<Tensor>

Multimodal

Multimodal outputs

Fields

text_logits: Option<Tensor>

image_logits: Option<Tensor>

text_embeddings: Option<Tensor>

image_embeddings: Option<Tensor>

Sequence

Sequence-to-sequence outputs

Fields

logits: Tensor

encoder_hidden_states: Option<Tensor>

decoder_hidden_states: Option<Tensor>

CLIP

CLIP contrastive outputs

Fields

logits_per_text: Tensor

logits_per_image: Tensor

text_embeds: Tensor

image_embeds: Tensor

Implementations

impl ModelOutputs

pub fn logits(logits: Tensor) -> Self

Create logits output

pub fn embeddings(embeddings: Tensor) -> Self

Create embeddings output

pub fn main_tensor(&self) -> &Tensor

Get main tensor output

Trait Implementations

impl Clone for ModelOutputs

fn clone(&self) -> ModelOutputs

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ModelOutputs

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

Formats the value using the given formatter.