下载
中文
注册

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()