/torchlayers

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

Primary LanguagePythonMIT LicenseMIT

Pinned issues

Issues