Struct Tensor 

pub struct Tensor { /* private fields */ }

A typed, shaped buffer of data.

Tensors hold raw bytes with a dtype and shape. Data is reference-counted so cloning a tensor is cheap (no copy). The actual bytes are only copied when mutated (copy-on-write).

Implementations

impl Tensor

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

Create a float32 tensor from a slice of f32 values.

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

Create a float64 tensor from a slice of f64 values.

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

Create a uint8 tensor (images, raw bytes).

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

Create an int32 tensor.

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

Create an int64 tensor (labels, indices).

pub fn bytes(data: Vec<u8>) -> Self

Create a raw bytes tensor (untyped).

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

Create a tensor from raw bytes, dtype, and shape.

Errors

Returns an error if the provided byte length does not match the product of shape elements multiplied by the element size for the given dtype.

pub fn dtype(&self) -> DType

The element type of this tensor.

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

The shape of this tensor.

pub fn num_elements(&self) -> Option<usize>

Total number of elements.

Returns None if the element count overflows usize.

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

Raw byte data (for zero-copy transfer to numpy/torch).

pub fn try_as_f32(&self) -> Result<&[f32]>

Interpret data as f32 slice.

Errors

Returns an error if the tensor’s dtype is not f32.

pub fn try_as_f64(&self) -> Result<&[f64]>

Interpret data as f64 slice.

Errors

Returns an error if the tensor’s dtype is not f64.

pub fn try_as_i32(&self) -> Result<&[i32]>

Interpret data as i32 slice.

Errors

Returns an error if the tensor’s dtype is not i32.

pub fn try_as_i64(&self) -> Result<&[i64]>

Interpret data as i64 slice.

Errors

Returns an error if the tensor’s dtype is not i64.

pub fn byte_len(&self) -> usize

Byte length of the data buffer.

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.