/torch-dimcheck

Dimensionality annotations for tensor parameters and return values

Primary LanguagePython