szymonmaszke / torchlayers

Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
570Updated 2 years ago

Alternatives and similar repositories for torchlayers:

Users that are interested in torchlayers are comparing it to the libraries listed below