PyTorch documentation¶
PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.
Notes
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.utils.bottleneck
- torch.utils.checkpoint
- torch.utils.cpp_extension
- torch.utils.data
- torch.utils.dlpack
- torch.utils.model_zoo
- torch.utils.tensorboard (experimental)
- torch.onnx
- torch.__config__
- torch.distributed.deprecated
torchvision Reference
