Notes
- Autograd mechanics
- Broadcasting semantics
- CUDA semantics
- Extending PyTorch
- Multiprocessing best practices
- Serialization semantics
Package Reference
torch
torch.Tensor
torch.sparse
torch.Storage
torch.nn
torch.nn.functional
torch.nn.init
torch.optim
torch.autograd
torch.distributions
torch.multiprocessing
torch.distributed
torch.legacy
torch.cuda
torch.utils.ffi
torch.utils.data
torch.utils.model_zoo
torch.onnx
torchvision Reference
- torchvision
torchvision.datasets
torchvision.models
torchvision.transforms
torchvision.utils
Indices and tables
- Index
- Module Index