Updated 6 months ago

https://github.com/szymonmaszke/torchlayers • Rank 7.7 • Science 10%

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks