About
The PyTorch Docstring skill automates the creation of professional, publication-ready documentation for Python codebases, specifically optimized for the PyTorch ecosystem. It ensures that all functions and methods include essential elements such as raw string formatting, Sphinx cross-references, LaTeX mathematical formulas for tensor shapes, and structured sections for arguments, return values, and examples. By strictly adhering to the patterns found in the core PyTorch library, this skill helps developers maintain consistency, improve code readability, and ensure that documentation renders perfectly in Sphinx-based web portals.
