torch.cuda.amp 表1 torch.cuda.amp 序号 API名称 npu对应API名称 是否支持 1 torch.cuda.amp.GradScaler torch.npu.amp.GradScaler 否 父主题： PyTorch1.5.0
