Struct SparseCSRTensor 

pub struct SparseCSRTensor { /* private fields */ }

Sparse tensor in CSR (Compressed Sparse Row) format.

CSR format is efficient for row-wise operations on 2D sparse matrices:

• crow_indices: shape [nrows + 1], dtype I64 — row pointers
• col_indices: shape [nnz], dtype I64 — column indices for each non-zero
• values: shape [nnz] — the non-zero values

For row i, the non-zeros are at positions crow_indices[i]..crow_indices[i+1] in col_indices and values.

Implementations

impl SparseCSRTensor

pub fn new( crow_indices: DenseI64Tensor, col_indices: DenseI64Tensor, values: DenseTensor, shape: [usize; 2], ) -> Result<Self, SparseTensorError>

Create a new sparse CSR tensor.

Arguments

• crow_indices - shape [nrows + 1], dtype I64
• col_indices - shape [nnz], dtype I64
• values - shape [nnz], any floating-point dtype
• shape - [nrows, ncols]

Errors

Returns error if shapes don’t match or indices are invalid.

pub fn id(&self) -> u64

pub fn crow_indices(&self) -> &DenseI64Tensor

pub fn col_indices(&self) -> &DenseI64Tensor

pub fn values(&self) -> &DenseTensor

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

pub fn nrows(&self) -> usize

pub fn ncols(&self) -> usize

pub fn nnz(&self) -> usize

pub fn dtype(&self) -> DType

pub fn device(&self) -> Device

pub fn version(&self) -> u64

pub fn to_dense(&self) -> Result<DenseTensor, SparseTensorError>

Convert this sparse CSR tensor to a dense tensor.

Trait Implementations

impl Clone for SparseCSRTensor

fn clone(&self) -> SparseCSRTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SparseCSRTensor

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

Formats the value using the given formatter.

impl PartialEq for SparseCSRTensor

fn eq(&self, other: &SparseCSRTensor) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for SparseCSRTensor