I want to improve documentation of my code to make dimensionality of tensors easily readable. Currently I’m doing this with a mix of of python type hints for dtypes and ‘manual’ code comments for shapes.
Now I came across this package that enables explicit shape-hints: https://github.com/ofnote/tsalib
My question: Is there any built-in functionality in pytorch that provides something similar? (Or is this planned in the future?)
What are best practices for annotating shapes?