Struct SparseTensorCSR 

pub struct SparseTensorCSR {
    pub values: Vec<f32>,
    pub col_indices: Vec<usize>,
    pub row_ptrs: Vec<usize>,
    pub shape: Vec<usize>,
    pub nnz: usize,
}

Sparse tensor in CSR (Compressed Sparse Row) format

Fields

values: Vec<f32>

Non-zero values

col_indices: Vec<usize>

Column indices

row_ptrs: Vec<usize>

Row pointers

shape: Vec<usize>

Shape of the tensor

nnz: usize

Number of non-zero elements

Implementations

impl SparseTensorCSR

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

Create a new sparse tensor in CSR format

pub fn from_coo(coo: &SparseTensorCOO) -> Result<Self>

Convert from COO format

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

Convert to COO format

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

Sparse matrix-vector multiplication (optimized for CSR)

Trait Implementations

impl Clone for SparseTensorCSR

fn clone(&self) -> SparseTensorCSR

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SparseTensorCSR

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

Formats the value using the given formatter.