.. PyTorch documentation master file, created by sphinx-quickstart on Fri Dec 23 13:31:47 2016. You can adapt this file completely to your liking, but it should at least contain the root `toctree` directive. :github_url: https://github.com/pytorch/pytorch 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 generally be no major performance limitations or gaps in documentation. We also expect to maintain backwards compatibility (although breaking changes can happen and notice will be given one release ahead of time). *Beta:* These features are tagged as Beta because the API may change based on user feedback, because the performance needs to improve, or because coverage across operators is not yet complete. For Beta features, we are committing 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 of binary distributions like PyPI or Conda, except sometimes behind run-time flags, and are at an early stage for feedback and testing. .. toctree:: :glob: :maxdepth: 1 :caption: Community community/* .. toctree:: :glob: :maxdepth: 1 :caption: Developer Notes notes/* .. toctree:: :maxdepth: 1 :caption: Language Bindings cpp_index Javadoc torch::deploy .. toctree:: :glob: :maxdepth: 2 :caption: Python API torch nn nn.functional tensors tensor_attributes tensor_view torch.amp torch.autograd torch.library cpu cuda torch.cuda.memory mps xpu mtia meta torch.backends torch.export torch.distributed torch.distributed.tensor torch.distributed.algorithms.join torch.distributed.elastic torch.distributed.fsdp torch.distributed.tensor.parallel torch.distributed.optim torch.distributed.pipelining torch.distributed.checkpoint torch.distributions torch.compiler torch.fft torch.func futures fx fx.experimental torch.hub torch.jit torch.linalg torch.monitor torch.signal torch.special torch.overrides torch.package profiler nn.init nn.attention onnx optim complex_numbers ddp_comm_hooks quantization rpc torch.random masked torch.nested size sparse storage torch.testing torch.utils torch.utils.benchmark torch.utils.bottleneck torch.utils.checkpoint torch.utils.cpp_extension torch.utils.data torch.utils.deterministic torch.utils.jit torch.utils.dlpack torch.utils.mobile_optimizer torch.utils.model_zoo torch.utils.tensorboard torch.utils.module_tracker type_info named_tensor name_inference torch.__config__ torch.__future__ logging torch_environment_variables .. toctree:: :maxdepth: 1 :caption: Libraries torchaudio TorchData TorchRec TorchServe torchtext torchvision PyTorch on XLA Devices Indices and tables ================== * :ref:`genindex` * :ref:`modindex`