PyTorch原生API

torch
torch.nn
torch.nn.functional
torch.Tensor
Tensor Attributes
Tensor Views
torch.autograd
torch.cuda
torch.cuda.amp
torch.distributed
torch.distributions
torch.fx
torch.linalg
torch.monitor
torch.special
torch.overrides
torch.package
torch.profiler
torch.nn.init
torch.onnx
torch.optim
torch.testing
torch.utils.benchmark
torch.utils.checkpoint
torch.utils.cpp_extension
torch.utils.tensorboard
Type Info
Named Tensors

父主题： PyTorch1.11.0