Struct PruningMask 

pub struct PruningMask {
    pub mask: Array2<f32>,
    pub pruned_weights: Option<Array2<f32>>,
    pub sparsity: f32,
}

Pruning mask for a weight tensor

Fields

mask: Array2<f32>

Binary mask (1 = keep, 0 = prune)

pruned_weights: Option<Array2<f32>>

Pruned weights (if keep_pruned_weights is enabled)

sparsity: f32

Sparsity ratio achieved

Implementations

impl PruningMask

pub fn new(mask: Array2<f32>) -> Self

Create a new pruning mask

pub fn apply(&self, weights: &Array2<f32>) -> Array2<f32>

Apply mask to weights

pub fn num_parameters(&self) -> usize

Count remaining (non-zero) parameters

pub fn compression_ratio(&self) -> f32

Get compression ratio

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.