PyTorch documentation

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

Notes

• Autograd mechanics
• Broadcasting semantics
• CPU threading and TorchScript inference
• CUDA semantics
• Extending PyTorch
• Frequently Asked Questions
• Features for large-scale deployments
• Multiprocessing best practices
• Reproducibility
• Serialization semantics
• Windows FAQ

Community

• PyTorch Contribution Guide
• PyTorch Governance
• PyTorch Governance | Persons of Interest

Package Reference

• torch
• torch.Tensor
• Tensor Attributes
• Type Info
• torch.sparse
• torch.cuda
• torch.Storage
• torch.nn
• torch.nn.functional
• torch.nn.init
• torch.optim
• torch.autograd
• torch.distributed
• torch.distributions
• torch.hub
• torch.jit
• torch.multiprocessing
• torch.random
• torch.utils.bottleneck
• torch.utils.checkpoint
• torch.utils.cpp_extension
• torch.utils.data
• torch.utils.dlpack
• torch.utils.model_zoo
• torch.utils.tensorboard
• torch.onnx
• torch.__config__

torchvision Reference

• torchvision
  • torchvision.datasets
  • torchvision.io
  • torchvision.models
  • torchvision.ops
  • torchvision.transforms
  • torchvision.utils

torchaudio Reference

• torchaudio

torchtext Reference

• torchtext

Indices and tables

• Index

• Module Index