Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
What is the szymonmaszke/torchlayers GitHub project? Description: "Shape and dimension inference (Keras-like) for PyTorch layers and neural networks". Written in Python. Explain what it does, its main use cases, key features, and who would benefit from using it.
Question is copied to clipboard — paste it after the AI opens.
Clone via HTTPS
Clone via SSH
Download ZIP
Download master.zipReport bugs or request features on the torchlayers issue tracker:
Open GitHub Issues