CeluV2

产品支持情况

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

功能说明

• 算子功能：激活函数Celu（Continuously Differentiable Exponential Linear Units）。
• 计算公式：

$$CELU(x)=\max (0,x)+\min (0,\alpha \ast (\exp (x/\alpha )-1))$$

其中：

• x: 输入张量

• alpha: CELU公式中的激活系数，默认值为1.0

参数说明

参数名	输入/输出	描述	数据类型	数据格式
x	输入	公式中的'x'，表示CELU激活函数的输入张量。支持空Tensor，支持非连续Tensor，维度支持0-8。	FLOAT16、FLOAT32、DT_BF16	ND
alpha	属性	CELU公式中的激活系数，默认值为1.0，不能为0。	FLOAT	-
y	输出	表示x经CELU计算得到的输出张量，shape和dtype与输入x一致。支持非连续Tensor。	FLOAT16、FLOAT32、DT_BF16	ND

约束说明

无。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_celu.cpp	通过aclnnCelu&aclnnInplaceCelu接口方式调用CeluV2算子。