Struct SparseLinear 

pub struct SparseLinear { /* private fields */ }

Sparse linear layer (also known as sparse fully connected layer)

Implementations

impl SparseLinear

pub fn new( in_features: usize, out_features: usize, sparsity: f32, use_bias: bool, ) -> TorshResult<Self>

Create a new sparse linear layer

pub fn from_weight(weight: CsrTensor, bias: Option<Tensor>) -> TorshResult<Self>

Create from existing sparse weight matrix

pub fn forward(&self, input: &Tensor) -> TorshResult<Tensor>

Forward pass

pub fn sparsity(&self) -> f32

Get weight sparsity

pub fn num_parameters(&self) -> usize

Get number of parameters (including bias)

pub fn structured_prune( &mut self, ratio: f32, dimension: usize, ) -> TorshResult<()>

Apply structured pruning (remove entire rows/columns)

pub fn magnitude_prune(&mut self, ratio: f32) -> TorshResult<()>

Apply magnitude-based unstructured pruning

Trait Implementations

impl Clone for SparseLinear

fn clone(&self) -> SparseLinear

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SparseLinear

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

Formats the value using the given formatter.