NanToNum

产品支持情况

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

功能说明

• 算子功能：将输入中的NaN、正无穷大和负无穷大值分别替换为nan、posinf、neginf指定的值。

• 计算公式：

$$y=\{\begin{array}{ll}nan& \text{if }x\text{ is NaN}\\ posinf& \text{if }x\text{ is }+\infty \\ neginf& \text{if }x\text{ is }-\infty \\ x& \text{otherwise}\end{array}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行nan_to_num计算的入参，公式中的x。	BFLOAT16、FLOAT16、FLOAT	ND
y	输出	待进行nan_to_num计算的出参，公式中的y。	BFLOAT16、FLOAT16、FLOAT	ND
nan	属性	替换tensor元素中NaN的值。	FLOAT	-
posinf	属性	替换tensor元素中正无穷大的值。	FLOAT	-
neginf	属性	替换tensor元素中负无穷大的值。	FLOAT	-

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_nan_to_num	通过aclnnNanToNum接口方式调用NanToNum算子。