FilesLast commitLast update
13 days ago
2 months ago
27 days ago
7 days ago
17 days ago
2 months ago
7 days ago
17 days ago
1 month ago
README

NanToNum

产品支持情况

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

功能说明

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

  • 计算公式:

y={nanif x is NaNposinfif x is +∞neginfif x is −∞xotherwisey=\begin{cases} 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{cases}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
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算子。