szymonmaszke/torchlayers

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

2 best points · 0 comments · 1 Hacker News post · 572 GitHub stars · View on GitHub

Hacker News discussions

• Torchlayers: Shape Inference for PyTorch (2 points, 0 comments, by jonbaer, 2020-03-29)