GitHub repository: yuezuegu/torchshape
Primary language: Python
Description: Calculates the output shape of Pytorch operations