文件最后提交记录最后更新时间
7 天前
25 天前
7 个月前
25 天前
25 天前
25 天前
7 个月前
8 天前
README

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阶跃函数,作为模型的激活函数。

  • 计算公式:

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

参数说明

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