Updated 6 months ago
https://github.com/szymonmaszke/torchlayers
Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
Shape and dimension inference (Keras-like) for PyTorch layers and neural networks