Heaviside

产品支持情况

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

功能说明

• 接口功能： 计算输入input中每个元素的Heaviside阶跃函数，作为模型的激活函数。

• 计算公式：

  $$\text{Heaviside}(\text{input},\text{values})=\{\begin{array}{ll}0,& \text{如果input}<0\\ \text{values},& \text{如果input}=0\\ 1,& \text{如果input}>0\end{array}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
input	输入	公式中的输入input。	BFLOAT16、FLOAT16、FLOAT	ND
values	输入	公式中的输入values。	BFLOAT16、FLOAT16、FLOAT	ND
out	输出	公式中的Heaviside。	BFLOAT16、FLOAT16、FLOAT	ND

• Atlas Kirin X90 处理器系列产品、Atlas Kirin 9030 处理器系列产品：不支持BFLOAT16。

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_heaviside	通过aclnnHeaviside接口方式调用Heaviside算子。
图模式调用	-	通过算子IR构图方式调用Heaviside算子。