文件最后提交记录最后更新时间
17 天前
2 天前
17 天前
17 天前
17 天前
17 天前
1 个月前
12 小时前
README

Relu6

产品支持情况

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

功能说明

  • 算子功能:激活函数Relu6,按元素对输入张量进行激活计算,将每个元素值限制在[0, 6]区间内。
  • 计算公式:

Relu6(x)=min⁡(max⁡(x,0),6)Relu6(x) = \min(\max(x, 0), 6)

等价于分段函数:

Relu6(x)={0,x<0x,0≤x≤66,x>6Relu6(x) = \begin{cases} 0, & x < 0 \\ x, & 0 \le x \le 6 \\ 6, & x > 6 \end{cases}

其中:

  • x: 输入张量

参数说明

参数名 输入/输出 描述 数据类型 数据格式
x 输入 公式中的'x',表示Relu6激活函数的输入张量。 FLOAT16、FLOAT32、INT32、DT_BF16 ND
y 输出 表示x经Relu6计算得到的输出张量,shape和dtype与输入x一致。 FLOAT16、FLOAT32、INT32、DT_BF16 ND

约束说明

  • 输入x和输出y的shape必须相同。
  • 最高支持8维张量。

调用说明

调用方式 样例代码 说明
图模式(GE IR) test_geir_relu6.cpp 通过GE IR图模式方式调用Relu6算子。