PyTorch原生API
torch
torch.nn
torch.nn.functional
torch.Tensor
Tensor Attributes
torch.amp
torch.autograd
torch.library
torch.cuda
torch.backends
torch.distributed
torch.distributed.algorithms.join
torch.distributed.elastic
torch.distributed.fsdp
torch.distributed.optim
torch.distributions
torch.futures
torch.fx
torch.hub
torch.jit
torch.linalg
torch.monitor
torch.special
torch.overrides
torch.package
torch.profiler
torch.nn.init
torch.onnx
torch.optim
torch.cpu
DDP Communication Hooks
Quantization
Distributed RPC Framework
torch.random
torch.Storage
torch.testing
torch.utils
torch.utils.benchmark
torch.utils.checkpoint
torch.utils.cpp_extension
torch.utils.data
torch.utils.mobile_optimizer
torch.utils.model_zoo
torch.utils.tensorboard
Type Info
Named Tensors
torch.__config__
父主题：
PyTorch2.2.0