Struct SparseMask 

pub struct SparseMask { /* private fields */ }

Sparse mask for weight matrices

Implementations

impl SparseMask

pub fn random(shape: &[usize], sparsity: f32) -> Result<Self>

Create a new sparse mask with random sparsity

pub fn from_magnitude(weights: &Tensor, threshold: f32) -> Result<Self>

Create mask from magnitude-based pruning

pub fn blocked( shape: &[usize], block_size: usize, sparsity: f32, ) -> Result<Self>

Create block-wise sparse mask

pub fn apply(&self, weights: &Tensor) -> Result<Tensor>

Apply mask to weights

pub fn nnz(&self) -> usize

Get number of non-zero elements

pub fn sparsity(&self) -> f32

Get sparsity level

pub fn mask(&self) -> &Tensor

Get the mask tensor

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.