torch_npu.npu.amp 表1 torch_npu.npu.amp APIAPI接口 说明 torch_npu.npu.amp.autocast Torch_npu提供amp相关的部分接口，具体可参考torch.amp。 torch_npu.npu.amp.GradScaler torch_npu.npu.amp.custom_fwd torch_npu.npu.amp.custom_bwd 父主题： Ascend Extension for PyTorch自定义API
