Struct CapturedTensor 

pub struct CapturedTensor {
    pub name: String,
    pub layer: Option<usize>,
    pub ne0: usize,
    pub ne1: usize,
    pub data: Vec<f32>,
}

Information about a single captured tensor.

Fields

name: String

The tensor name (e.g. "l_out-13").

layer: Option<usize>

The layer index extracted from the name, or None if the name doesn’t follow the "prefix-N" pattern.

ne0: usize

First dimension (typically n_embd / hidden dimension).

ne1: usize

Second dimension (typically n_tokens).

data: Vec<f32>

Flattened f32 data with ne0 * ne1 elements.

Layout (row-major from ggml’s perspective): data[token_idx * ne0 + dim_idx]

This matches the ggml tensor layout where ne[0] is the innermost (contiguous) dimension.

Implementations

impl CapturedTensor

pub fn n_embd(&self) -> usize

Number of embedding dimensions (alias for ne0).

pub fn n_tokens(&self) -> usize

Number of tokens (alias for ne1).

pub fn token_embedding(&self, token_idx: usize) -> Option<&[f32]>

Get the hidden state for a specific token.

Returns a slice of n_embd floats, or None if token_idx is out of range.

Trait Implementations

impl Clone for CapturedTensor

fn clone(&self) -> CapturedTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CapturedTensor

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

Formats the value using the given formatter.