Struct TensorSnapshot 

pub struct TensorSnapshot {
    pub name: String,
    pub shape: Vec<usize>,
    pub data: Vec<f32>,
}

A named, flat snapshot of a single tensor.

Fields

name: String

Tensor name / checkpoint key.

shape: Vec<usize>

Shape dimensions, e.g. [128, 256] for a 2-D tensor.

data: Vec<f32>

Flattened row-major element data.

Implementations

impl TensorSnapshot

pub fn new(name: impl Into<String>, shape: Vec<usize>, data: Vec<f32>) -> Self

Create a new snapshot.

pub fn numel(&self) -> usize

Number of elements: product of shape dimensions, or data.len() when shape is empty (scalar / rank-0 tensors).

pub fn is_compatible(&self, other: &TensorSnapshot) -> bool

Returns true when self and other have identical shapes and can therefore be compared element-wise.

Trait Implementations

impl Clone for TensorSnapshot

fn clone(&self) -> TensorSnapshot

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorSnapshot

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

Formats the value using the given formatter.