文件最后提交记录最后更新时间
29 天前
29 天前
29 天前
2 个月前
10 天前
6 个月前
12 天前
README

ReluV2

产品支持情况

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

功能说明

  • 算子功能:激活函数,返回与输入tensor shape相同的tensor,tensor中value大于等于0时,取该value,小于0,取0。

  • 计算公式:

reluv2(self)={self,self>00,self≤0relu_v2(self) = \begin{cases} self, & self\gt 0 \\ 0, & self\le 0 \end{cases}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 待进行ReluV2计算的入参,公式中的输入self。 FLOAT、FLOAT16、BFLOAT16、INT8、INT32、UINT8 ND
y 输出 进行ReluV2计算后的输出,公式中的relu_v2(self)。 FLOAT、FLOAT16、INT8、INT32、INT64、BFLOAT16、UINT8 ND
mask 输出 进行ReluV2计算后的输出,当输入的值大于0时,该值为1,当输入的值小于等于0时,该值为0。 UINT1 ND

约束说明

调用说明

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