Enum SparseTensor 

pub enum SparseTensor {
    COO(SparseCOO),
    CSR(SparseCSR),
}

Unified sparse tensor interface supporting multiple formats.

Variants

COO(SparseCOO)

COO format

CSR(SparseCSR)

CSR format

Implementations

impl SparseTensor

pub fn from_coo( indices: Vec<Vec<usize>>, values: Vec<f32>, shape: Vec<usize>, ) -> Self

Creates a sparse tensor from COO data.

pub fn from_coords( coords: &[(usize, usize)], values: &[f32], shape: &[usize], ) -> Self

Creates a 2D sparse tensor from coordinate list.

pub fn from_dense(tensor: &Tensor<f32>, threshold: f32) -> Self

Creates from a dense tensor, keeping only non-zero elements.

pub fn eye(n: usize) -> Self

Creates an identity matrix in sparse format.

pub fn diag(values: &[f32]) -> Self

Creates a sparse diagonal matrix.

pub fn nnz(&self) -> usize

Returns number of non-zero elements.

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

Returns the shape.

pub fn density(&self) -> f32

Returns the density.

pub fn to_dense(&self) -> Tensor<f32>

Converts to dense tensor.

pub fn to_csr(&self) -> SparseCSR

Converts to CSR format.

pub fn to_coo(&self) -> SparseCOO

Converts to COO format.

pub fn matvec(&self, x: &[f32]) -> Vec<f32>

Sparse matrix-vector multiplication.

pub fn matmul(&self, dense: &Tensor<f32>) -> Tensor<f32>

Sparse-dense matrix multiplication.

pub fn mul_scalar(&self, scalar: f32) -> Self

Element-wise multiplication with a scalar.

Trait Implementations

impl Clone for SparseTensor

fn clone(&self) -> SparseTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SparseTensor

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

Formats the value using the given formatter.