Enum ModelSpecificValue 

pub enum ModelSpecificValue {
    Tensor {
        data: Vec<f32>,
        shape: Vec<usize>,
    },
    IntTensor {
        data: Vec<i64>,
        shape: Vec<usize>,
    },
    UintTensor {
        data: Vec<u32>,
        shape: Vec<usize>,
    },
    Int(i64),
    Float(f64),
    IntVec(Vec<i64>),
    UintVec(Vec<u32>),
    FloatVec(Vec<f32>),
    TupleVec(Vec<(u32, u32)>),
    Bool(bool),
}

Model-specific output values that vary by architecture.

Different vision models require different auxiliary outputs beyond pixel_values. This enum captures the common types of such outputs.

Variants

Tensor

A tensor with shape information (data as flat vec, shape as dims)

Fields

data: Vec<f32>

shape: Vec<usize>

IntTensor

A tensor of integers (e.g., aspect_ratio_ids)

Fields

data: Vec<i64>

shape: Vec<usize>

UintTensor

A tensor of unsigned integers (e.g., image_grid_thw)

Fields

data: Vec<u32>

shape: Vec<usize>

Int(i64)

Simple integer value

Float(f64)

Simple float value

IntVec(Vec<i64>)

List of integers

UintVec(Vec<u32>)

List of unsigned integers

FloatVec(Vec<f32>)

List of floats

TupleVec(Vec<(u32, u32)>)

List of tuples (e.g., image sizes)

Bool(bool)

Boolean flag

Implementations

impl ModelSpecificValue

pub fn uint_1d(data: Vec<u32>) -> Self

Create a 1D uint tensor from a vector.

pub fn uint_2d(data: Vec<u32>, rows: usize, cols: usize) -> Self

Create a 2D uint tensor.

pub fn int_1d(data: Vec<i64>) -> Self

Create a 1D int tensor from a vector.

pub fn int_2d(data: Vec<i64>, rows: usize, cols: usize) -> Self

Create a 2D int tensor.

pub fn first_dim(&self) -> Option<usize>

Get the first dimension of a tensor variant, if applicable.

Trait Implementations

impl Clone for ModelSpecificValue

fn clone(&self) -> ModelSpecificValue

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModelSpecificValue

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

Formats the value using the given formatter.