NonFiniteCheck

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品 ×
Atlas 训练系列产品 ×
Kirin X90 处理器系列产品
Kirin 9030 处理器系列产品

功能说明

  • 算子功能:用于检查一个包含多个张量的列表tensor_list中是否存在非有限数(+inf,-inf,NaN)。如果存在非有限数,则将found_flag设置为1,否则设置为0。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
tensor_list 输入 被检查的张量列表 FLOAT、FLOAT16、BFLOAT16 ND
found_flag 输出 用于指示`tensor_list`中是否存在非有限数 FLOAT ND
  • Kirin X90/Kirin 9030 处理器系列产品: 不支持BFLOAT16。

约束说明

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_non_finite_check 通过算子IR构图方式调用NonFiniteCheck算子。