Struct SparseMask 

pub struct SparseMask {
    pub mask: Vec<bool>,
}

Boolean pruning mask over a weight tensor.

true → weight is active (kept). false → weight is pruned (zeroed).

Fields

mask: Vec<bool>

Element-wise mask (same length as the weight tensor).

Implementations

impl SparseMask

pub fn all_active(n: usize) -> Self

Create a mask with all weights active (unpruned).

pub fn all_pruned(n: usize) -> Self

Create a mask with all weights pruned.

pub fn len(&self) -> usize

Number of elements in the mask.

pub fn is_empty(&self) -> bool

Returns true if the mask covers an empty weight tensor.

pub fn count_active(&self) -> usize

Number of active (non-pruned) weights.

pub fn count_pruned(&self) -> usize

Number of pruned weights.

pub fn sparsity(&self) -> f32

Fraction of weights that are pruned ∈ [0, 1].

pub fn apply(&self, weights: &[f32]) -> Vec<f32>

Apply the mask to a weight slice: pruned weights become 0.

Returns a new Vec<f32> of the same length.

Panics

Panics if weights.len() != self.len().

pub fn apply_in_place(&self, weights: &mut [f32])

Apply the mask in-place: pruned weights become 0.

Panics

Panics if weights.len() != self.len().

pub fn and(&self, other: &Self) -> Self

Combine two masks with logical AND (both must be active to stay active).

Panics

Panics if the masks have different lengths.

pub fn or(&self, other: &Self) -> Self

Combine two masks with logical OR (at least one active → active).

Panics

Panics if the masks have different lengths.

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.