szymonmaszke / torchlayers

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

Related projects

Alternatives and complementary repositories for torchlayers