Struct Tensor 

pub struct Tensor { /* private fields */ }

A multi-dimensional tensor stored as raw bytes.

Implementations

impl Tensor

pub fn new(data: Vec<u8>, shape: Vec<usize>, dtype: DType) -> Self

Create a new tensor from raw bytes.

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

Create an F32 tensor from a float slice.

pub fn zeros_f32(shape: Vec<usize>) -> Self

Create a zero-filled F32 tensor.

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

pub fn dtype(&self) -> DType

pub fn numel(&self) -> usize

Total number of logical elements.

pub fn size_bytes(&self) -> usize

Size of the underlying data in bytes.

pub fn data(&self) -> &[u8]

Raw byte slice of the tensor data.

pub fn data_mut(&mut self) -> &mut [u8]

Mutable raw byte slice of the tensor data.

pub fn as_f32_slice(&self) -> &[f32]

View the data as an f32 slice. Panics if dtype is not F32.

pub fn as_f32_slice_mut(&mut self) -> &mut [f32]

View the data as a mutable f32 slice. Panics if dtype is not F32.

pub fn to_f32_vec(&self) -> Vec<f32>

Convert tensor data to a Vec<f32>, dequantizing if necessary.