Struct PruningMask 

pub struct PruningMask {
    pub mask: Vec<bool>,
    pub shape: Vec<usize>,
    pub name: Option<String>,
}

Pruning mask indicating which weights are kept or pruned

The mask uses true to indicate weights that should be KEPT, and false for weights that should be PRUNED (zeroed).

Fields

mask: Vec<bool>

Boolean mask (true = keep, false = prune)

shape: Vec<usize>

Shape of the mask (same as weight tensor)

name: Option<String>

Optional layer name

Implementations

impl PruningMask

pub fn new(mask: Vec<bool>, shape: Vec<usize>) -> Self

Creates a new pruning mask

pub fn with_name(mask: Vec<bool>, shape: Vec<usize>, name: String) -> Self

Creates a new pruning mask with a name

pub fn ones(shape: &[usize]) -> Self

Creates an all-ones mask (keep everything)

pub fn zeros(shape: &[usize]) -> Self

Creates an all-zeros mask (prune everything)

pub fn numel(&self) -> usize

Returns the number of elements in the mask

pub fn num_kept(&self) -> usize

Returns the number of weights kept (not pruned)

pub fn num_pruned(&self) -> usize

Returns the number of weights pruned

pub fn sparsity(&self) -> f32

Returns the sparsity (fraction of pruned weights)

pub fn and(&self, other: &PruningMask) -> Result<PruningMask>

Combines two masks with logical AND (both must keep)

Errors

Returns an error if mask sizes don’t match

pub fn or(&self, other: &PruningMask) -> Result<PruningMask>

Combines two masks with logical OR (either keeps)

Errors

Returns an error if mask sizes don’t match

pub fn invert(&self) -> PruningMask

Inverts the mask

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

Applies the mask to a weight tensor, zeroing pruned weights

Errors

Returns an error if sizes don’t match

Trait Implementations

impl Clone for PruningMask

fn clone(&self) -> PruningMask

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PruningMask

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

Formats the value using the given formatter.