文件最后提交记录最后更新时间
30 天前
30 天前
30 天前
3 个月前
3 个月前
3 个月前
README

Selu

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

  • 算子功能:对输入 Tensor 逐元素计算 SELU(Scaled Exponential Linear Unit)激活函数。

  • 计算公式:

    SELU(x)=scale×(max⁡(0,x)+min⁡(0,α×(ex−1)))\text{SELU}(x) = \text{scale} \times \left( \max(0, x) + \min\left(0, \alpha \times \left(e^x - 1\right)\right) \right)

    其中:

    • α=1.6732632423543772848170429916717\alpha = 1.6732632423543772848170429916717

    • scale=1.0507009873554804934193349852946\text{scale} = 1.0507009873554804934193349852946

    • 分段行为:

      • x>0x > 0 时:SELU(x)=scale×x\text{SELU}(x) = \text{scale} \times x(线性区)
      • x≤0x \le 0 时:SELU(x)=scale×α×(ex−1)\text{SELU}(x) = \text{scale} \times \alpha \times (e^x - 1)(指数饱和区)

参数说明

参数名 输入/输出 描述 数据类型 数据格式
x 输入 公式中的输入x,支持1D-8D Tensor。 FLOAT、FLOAT16、BFLOAT16、INT32、INT8 ND
y 输出 公式中的输出y,shape与输入x相同。 FLOAT、FLOAT16、BFLOAT16、INT32、INT8 ND

约束说明