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