Struct SparseMask 

pub struct SparseMask {
    pub size: usize,
    pub row_ptr: Vec<usize>,
    pub col_idx: Vec<usize>,
    pub values: Vec<f32>,
}

Sparse mask in CSR (Compressed Sparse Row) format.

Efficient storage for sparse attention masks.

Fields

size: usize

Number of rows/columns

row_ptr: Vec<usize>

Row pointers (size + 1 elements)

col_idx: Vec<usize>

Column indices

values: Vec<f32>

Non-zero values

Implementations

impl SparseMask

pub fn from_dense(dense: &[Vec<f32>], threshold: f32) -> Self

Create from dense mask, keeping values above threshold.

pub fn from_toroidal(mask: &ToroidalMask, threshold: f32) -> Self

Create from ToroidalMask with threshold.

pub fn nnz(&self) -> usize

Number of non-zero elements.

pub fn sparsity(&self) -> f32

Sparsity ratio (1.0 = fully sparse, 0.0 = fully dense).

pub fn get(&self, i: usize, j: usize) -> f32

Get value at (i, j), or 0 if not stored.

pub fn memory_bytes(&self) -> usize

Memory usage in bytes (approximate).

Trait Implementations

impl Clone for SparseMask

fn clone(&self) -> SparseMask

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SparseMask

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

Formats the value using the given formatter.