CANN/ops-nn Elu算子实现

Elu

【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn

产品支持情况

功能说明

• 接口功能: 对输入张量self中的每个元素x调用指数线性单元激活函数ELU，并将得到的结果存入输出张量out中。

• 计算公式：

  $$ ELU(x) = \begin{cases} scale \ast x, \quad x > 0\ \alpha \ast scale \ast (exp(x \ast inputScale)-1), \quad x \leq 0 \end{cases} $$

• 参数说明：

  参数名	输入/输出	描述	使用说明	数据类型	数据格式	维度(shape)	非连续Tensor
  self	输入	表示ELU激活函数的输入，公式中的x。	支持空Tensor。	FLOAT、FLOAT16、BFLOAT16	ND	0-8	√
  alpha	输入	表示ELU激活函数的激活系数，公式中的α。	数据类型需要是可转换为FLOAT的数据类型。	-	-	-	-
  scale	输入	表示ELU激活函数的缩放系数，公式中的scale。	数据类型需要是可转换为FLOAT的数据类型。	-	-	-	-
  inputScale	输入	表示ELU激活函数的输入的缩放系数，公式中的inputScale。	数据类型需要是可转换为FLOAT的数据类型。	-	-	-	-
  out	输出	表示ELU激活函数的输出。	shape需要与self一致。	FLOAT、FLOAT16、BFLOAT16	ND	0-8	√

  - Atlas 训练系列产品 ：数据类型支持FLOAT、FLOAT16。

约束说明

无

调用说明

【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn