PyTorch2.0
torch
torch.nn
torch.nn.functional
torch.Tensor
Tensor Attributes
torch.autograd
torch.library
torch.cuda
torch.cuda.amp
torch.mps
torch.backends
torch.distributed
torch.distributed.algorithms.join
torch.distributed.elastic
torch.distributed.fsdp
torch.distributed.optim
torch.distributed.tensor.parallel
torch.distributed.checkpoint
torch.distributions
torch._dynamo
torch.fft
torch.func
torch.futures
torch.fx
torch.hub
torch.jit
torch.linalg
torch.monitor
torch.signal
torch.special
torch.overrides
torch.package
torch.profiler
torch.nn.init
torch.onnx
torch.onnx.diagnostics
torch.optim
DDP Communication Hooks
Pipeline Parallelism
Quantization
Distributed RPC Framework
torch.random
torch.nested
torch.sparse
torch.Storage
torch.testing
torch.utils.benchmark
torch.utils.checkpoint
torch.utils.cpp_extension
torch.utils.data
torch.utils.dlpack
torch.utils.mobile_optimizer
torch.utils.model_zoo
torch.utils.tensorboard
Type Info
Named Tensors
torch.__config__
父主题：
PyTorch原生API