Skip to main content

sparsify

entrenar::aprender_compat::pruning

Function sparsify 

Source 
pub fn sparsify(
    tensor: &Tensor,
    mask: &SparsityMask,
    format: SparseFormat,
) -> Result<SparseTensor, PruningError>
Expand description

Apply a sparsity mask to a tensor and return sparse representation.

§Arguments

  • tensor - Dense tensor to sparsify
  • mask - Binary sparsity mask (1 = keep, 0 = prune)
  • format - Target sparse format

§Returns

Sparse tensor in requested format