.. 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`