Struct Tensor 

pub struct Tensor { /* private fields */ }

Dense tensor with arbitrary dimensions.

Row-major (C-order) storage. Shape [d0, d1, ..., dk] means element [i0, i1, ..., ik] is at offset i0*stride[0] + i1*stride[1] + ....

Implementations

impl Tensor

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

Create a new tensor with the given shape and data.

Errors

Returns error if data length doesn’t match shape product.

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

Create a zero tensor with the given shape.

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

Tensor shape.

pub fn ndim(&self) -> usize

Number of dimensions (rank).

pub fn len(&self) -> usize

Total number of elements.

pub fn is_empty(&self) -> bool

Whether the tensor is empty.

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

Raw data slice.

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

Mutable data slice.

pub fn get(&self, indices: &[usize]) -> f32

Get element at multi-index.

pub fn set(&mut self, indices: &[usize], value: f32)

Set element at multi-index.

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

Reshape tensor (must have same total elements).

Errors

Returns error if new shape has different total elements.

pub fn transpose(&self, perm: &[usize]) -> Self

Transpose: permute dimensions according to perm.

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.