PyTorch1.11.0
torch
torch.nn
torch.nn.functional
torch.Tensor
Tensor Attributes
Tensor Views
torch.autograd
torch.cuda
torch.cuda.amp
torch.backends
torch.distributed
torch.distributed.algorithms.join
torch.distributed.elastic
torch.distributed.fsdp
torch.distributed.optim
torch.distributions
torch.fft
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
DDP Communication Hooks
Pipeline Parallelism
Quantization
Distributed RPC Framework
torch.random
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
父主题:
PyTorch原生API