torch_npu.npu.amp

表1 torch_npu.npu.amp API
API接口	说明
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