Struct Tensor 

pub struct Tensor { /* private fields */ }

A multi-dimensional tensor with typed storage

Implementations

impl Tensor

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

Create a new tensor from raw data bytes with the given shape and dtype

pub unsafe fn from_storage( storage: TensorStorage, shape: Vec<usize>, dtype: DType, offset: usize, ) -> Result<Self, TensorError>

Create a tensor from existing storage

Safety

The storage must contain valid data for the given shape and dtype. The offset + size must not exceed the storage length.

pub fn zeros(shape: Vec<usize>, dtype: DType) -> Self

Create a tensor filled with zeros

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

Create a tensor from f32 data

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

Get the shape of the tensor

pub fn ndim(&self) -> usize

Get the number of dimensions

pub fn numel(&self) -> usize

Get the total number of elements

pub fn dtype(&self) -> DType

Get the data type

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

Get the strides

pub fn name(&self) -> Option<&str>

Get the tensor name (for GPU weight lookup)

pub fn set_name(&mut self, name: impl Into<String>)

Set the tensor name (for GPU weight lookup)

pub fn with_name(self, name: impl Into<String>) -> Self

Create a named tensor (builder pattern)

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

Get the raw byte data

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

Get mutable access to the raw byte data

pub fn as_f32(&self) -> Result<&[f32], TensorError>

Get the data as f32 slice (only valid for F32 dtype)

pub fn as_f32_mut(&mut self) -> Result<&mut [f32], TensorError>

Get mutable access to data as f32 slice (only valid for F32 dtype)

pub fn is_contiguous(&self) -> bool

Check if the tensor is contiguous in memory

pub fn contiguous(&self) -> Result<Self, TensorError>

Return a contiguous copy of this tensor if not already contiguous

pub fn reshape(&self, new_shape: Vec<usize>) -> Result<Self, TensorError>

Reshape the tensor to a new shape

Returns a new tensor with the same data but different shape. The new tensor has its own copy of the data to allow mutation.

Trait Implementations

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Tensor

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

Formats the value using the given formatter.