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.distributed.tensor.parallel
torch.distributed.checkpoint
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
torch.cpu
torch.compiler
torch.export
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.4.0