shape-commentator (pypi): You can easily add numpy.ndarray.shape information to your script as comments.
Keywords: numpy|ndarray|shape comment|tool|python|pytorch