PyTorch documentation

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

Notes

• Autograd mechanics
• Broadcasting semantics
• CUDA semantics
• Extending PyTorch
• Frequently Asked Questions
• Multiprocessing best practices
• Reproducibility
• Serialization semantics
• Windows FAQ

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.jit
• torch.multiprocessing
• torch.utils.bottleneck
• torch.utils.checkpoint
• torch.utils.cpp_extension
• torch.utils.data
• torch.utils.dlpack
• torch.utils.ffi
• torch.hub
• torch.utils.model_zoo
• torch.onnx
• torch.distributed.deprecated
• torch.legacy

Indices and tables

• Index

• Module Index