mirror of
https://github.com/MLSysBook/TinyTorch.git
synced 2026-04-28 22:47:31 -05:00
3741e9c6ef
Stage 4 of TinyTorch API simplification: - Added flatten() and max_pool2d() helper functions - Renamed MultiChannelConv2D to Conv2d for PyTorch compatibility - Updated Conv2d to inherit from Module base class - Use Parameter() for weights and bias with automatic registration - Added backward compatibility alias: MultiChannelConv2D = Conv2d - Updated all test code to use Conv2d - Exported changes to tinytorch.core.spatial API now provides PyTorch-like spatial operations while maintaining educational value of implementing core convolution algorithms.