(beta)torch_npu.npu.get_npu_overflow_flag
接口原型
torch_npu.npu.get_npu_overflow_flag()
功能描述
检测npu计算过程中是否有数值溢出。
支持的型号
- Atlas 训练系列产品
调用示例
>>>a = torch.Tensor([65535]).npu().half() >>>a = a + a >>>ret = torch_npu.npu.get_npu_overflow_flag()
父主题: torch_npu.npu