szymonmaszke / torchlayers

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

szymonmaszke/torchlayers Issues