Celu
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
功能说明
-
算子功能:激活函数Celu(Continuously Differentiable Exponential Linear Units)。
-
计算公式:
CELU(x)=max(0,x)+min(0,α∗(exp(x/α)−1))\text{CELU}(x) = \max(0,x) + \min(0, \alpha * (\exp(x/\alpha) - 1))
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 输入张量。 | 见下方 | ND |
| alpha | 属性 | CELU公式中的α\alpha值,默认值为1.0。 | FLOAT | - |
| z | 输出 | 输出张量,shape和dtype与输入x一致。 | 同self | ND |
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:数据类型支持FLOAT、FLOAT16。
约束说明
无
调用说明
| 调用方式 | 样例代码 | 说明 |
|---|---|---|
| aclnn接口 | test_celu | 通过[aclnnCelu]接口方式调用Celu算子。 |