PyTorch documentation

PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.

Features described in this documentation are classified by release status:

Stable: These features will be maintained long-term and there should generallybe no major performance limitations or gaps in documentation.We also expect to maintain backwards compatibility (althoughbreaking changes can happen and notice will be given one release aheadof time).

Beta: These features are tagged as Beta because the API may change based onuser feedback, because the performance needs to improve, or becausecoverage across operators is not yet complete. For Beta features, we arecommitting to seeing the feature through to the Stable classification.We are not, however, committing to backwards compatibility.

Prototype: These features are typically not available as part ofbinary distributions like PyPI or Conda, except sometimes behind run-timeflags, and are at an early stage for feedback and testing.

Language Bindings

Python API

Indices and tables