Struct SparseCSR 

pub struct SparseCSR {
    pub row_ptr: Vec<usize>,
    pub col_indices: Vec<usize>,
    pub values: Vec<f32>,
    pub shape: Vec<usize>,
}

Sparse tensor in CSR (Compressed Sparse Row) format.

Efficient for row-wise operations and sparse matrix-vector products.

Fields

row_ptr: Vec<usize>

Row pointers (length = nrows + 1)

col_indices: Vec<usize>

Column indices for each non-zero

values: Vec<f32>

Values for each non-zero

shape: Vec<usize>

Shape [nrows, ncols]

Implementations

impl SparseCSR

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

Creates a new CSR sparse matrix.

pub fn nnz(&self) -> usize

Returns number of non-zero elements.

pub fn nrows(&self) -> usize

Returns number of rows.

pub fn ncols(&self) -> usize

Returns number of columns.

pub fn density(&self) -> f32

Returns the density.

pub fn row(&self, row_idx: usize) -> impl Iterator<Item = (usize, f32)> + '_

Gets entries for a specific row.

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

Sparse matrix-vector multiplication: A @ x.

pub fn matmul_dense(&self, b: &Tensor<f32>) -> Tensor<f32>

Sparse matrix-matrix multiplication: A @ B (where B is dense).

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

Converts to dense tensor.

pub fn to_coo(&self) -> SparseCOO

Converts to COO format.

Trait Implementations

impl Clone for SparseCSR

fn clone(&self) -> SparseCSR

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SparseCSR

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

Formats the value using the given formatter.