torsh-nn 0.1.2

Neural network modules for ToRSh with PyTorch-compatible API
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

//! Pooling functional operations
//!
//! This module contains functional pooling operations (max pool, avg pool, adaptive pool,
//! padding operations, etc.), split from the original monolithic pooling.rs.

// Main pooling functions
pub mod functions;

// Tests
#[cfg(test)]
mod functions_2;

// Re-export all public functions
pub use functions::*;