Struct MockTensor 

pub struct MockTensor { /* private fields */ }

A mock tensor that stores shape and optional f32 data. No GPU, no Candle — pure Rust.

Implementations

impl MockTensor

pub fn zeros(shape: &[usize], dtype: DataType) -> Self

Create a zero-filled tensor with given shape.

pub fn from_f32(data: Vec<f32>, shape: &[usize]) -> Self

Create a tensor from f32 data.

pub fn from_u32(data: &[u32], shape: &[usize]) -> Self

Create a tensor from u32 token IDs (stored as f32 internally).

pub fn into_ref(self) -> TensorRef

Wrap as TensorRef.

Trait Implementations

impl Clone for MockTensor

fn clone(&self) -> MockTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MockTensor

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

Formats the value using the given formatter.

impl TensorLike for MockTensor

fn as_any(&self) -> &dyn Any

Downcast support for backend-specific fast paths

fn shape(&self) -> &[usize]

Get tensor shape

fn dtype(&self) -> DataType

Get tensor data type

fn device(&self) -> Device

Get device where tensor resides

fn is_contiguous(&self) -> bool

Check if tensor is contiguous in memory

fn view(&self, start: &[usize], end: &[usize]) -> Result<TensorRef>

Create a view/slice of this tensor

fn reshape(&self, shape: &[usize]) -> Result<TensorRef>

Reshape tensor to new shape (must have same number of elements)

fn to_cpu(&self) -> Result<TensorRef>

Convert tensor to CPU device

fn to_device(&self, _device: &Device) -> Result<TensorRef>

Convert tensor to specific device

fn to_dtype(&self, dtype: DataType) -> Result<TensorRef>

Convert tensor to specific data type

fn to_vec_f32(&self) -> Result<Vec<f32>>

Extract tensor data as Vec (for logits sampling) This is a convenience method for backends that need to extract data

fn to_vec_u32(&self) -> Result<Vec<u32>>

Extract tensor data as Vec (for token IDs) This is a convenience method for backends that need to extract token data

fn argmax_last_dim_u32(&self) -> Result<u32>

Fast path: argmax over the last dimension, returning the selected token id.

fn numel(&self) -> usize

Get total number of elements

fn ndim(&self) -> usize

Get number of dimensions

fn is_scalar(&self) -> bool

Check if tensor is scalar (0-dimensional)

fn size_bytes(&self) -> usize

Get size in bytes for this tensor