Struct SparseTensorCOO 

pub struct SparseTensorCOO {
    pub values: Vec<f32>,
    pub rows: Vec<usize>,
    pub cols: Vec<usize>,
    pub shape: Vec<usize>,
    pub nnz: usize,
}

Sparse tensor in COO (Coordinate) format

Fields

values: Vec<f32>

Non-zero values

rows: Vec<usize>

Row indices

cols: Vec<usize>

Column indices

shape: Vec<usize>

Shape of the tensor

nnz: usize

Number of non-zero elements

Implementations

impl SparseTensorCOO

pub fn new( values: Vec<f32>, rows: Vec<usize>, cols: Vec<usize>, shape: Vec<usize>, ) -> Result<Self>

Create a new sparse tensor in COO format

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

Create sparse tensor from dense tensor (threshold-based)

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

Convert to dense tensor

pub fn spmv(&self, vec: &Tensor) -> Result<Tensor>

Sparse matrix-vector multiplication

pub fn spmm(&self, other: &SparseTensorCOO) -> Result<SparseTensorCOO>

Sparse matrix-matrix multiplication

pub fn add(&self, other: &SparseTensorCOO) -> Result<SparseTensorCOO>

Element-wise addition with another sparse tensor

pub fn transpose(&self) -> Result<SparseTensorCOO>

Transpose the sparse matrix

pub fn sparsity(&self) -> f32

Get sparsity ratio (fraction of zero elements)

Trait Implementations

impl Clone for SparseTensorCOO

fn clone(&self) -> SparseTensorCOO

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SparseTensorCOO

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

Formats the value using the given formatter.