torsh-sparse 0.1.2

Sparse tensor operations for ToRSh with SciRS2 integration
Documentation
1
2
3
4
5
6
7
8
9
10
//! Sparse neural network layers

pub mod embedding;
pub mod linear;

// Re-export commonly used layer types
pub use embedding::{SparseEmbedding, SparseEmbeddingStats};
pub use linear::SparseMemoryStats;
// Import SparseLinear from the main layers module that has pruning methods
pub use crate::layers::linear::SparseLinear;