Struct PruningMetadata 

pub struct PruningMetadata { /* private fields */ }

Pruning metadata tracking which elements/blocks are pruned

Implementations

impl PruningMetadata

pub fn new( strategy: PruningStrategy, original_shape: Shape, achieved_sparsity: f32, ) -> Self

Create new pruning metadata

pub fn with_indices(self, indices: Vec<usize>) -> Self

Set pruned indices for unstructured pruning

pub fn with_blocks(self, blocks: Vec<(usize, usize)>) -> Self

Set pruned blocks for structured pruning

pub fn with_channels(self, channels: Vec<usize>) -> Self

Set pruned channels

pub fn with_threshold(self, threshold: f32) -> Self

Set threshold value

pub fn strategy(&self) -> PruningStrategy

Get pruning strategy

pub fn sparsity(&self) -> f32

Get achieved sparsity

pub fn compression_ratio(&self) -> f32

Get compression ratio

pub fn num_pruned_elements(&self) -> usize

Get number of pruned elements

pub fn is_element_pruned(&self, index: usize) -> bool

Check if an element is pruned

pub fn memory_savings(&self, dtype: DType) -> usize

Get memory savings in bytes

Trait Implementations

impl Clone for PruningMetadata

fn clone(&self) -> PruningMetadata

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PruningMetadata

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

Formats the value using the given formatter.