PyTorch1.8.1
torch
torch.nn
torch.nn.functional
torch.distributed
torch.cuda
torch.cuda.amp
NPU自定义算子
亲和库
父主题：
支持PyTorch API清单