mindspore.nn.ELU

class mindspore.nn.ELU(alpha=1.0)[源代码]

指数线性单元激活函数（Exponential Linear Unit activation function）。

对输入的每个元素计算ELU。该激活函数定义如下：

\[E_{i} = \begin{cases} x_i, &\text{if } x_i \geq 0; \cr \alpha * (\exp(x_i) - 1), &\text{otherwise.} \end{cases}\]

其中，\(x_i\) 表示输入的元素，\(\alpha\) 表示 alpha 参数。

ELU相关图参见 ELU 。

参数：

  • alpha (float) – ELU的alpha值，数据类型为浮点数。默认值：1.0。

输入：

  • x （Tensor） - 用于计算ELU的任意维度的Tensor，数据类型为float16或float32。

输出：

Tensor，数据类型和shape与 x 相同。

异常：

  • TypeError - alpha 不是浮点数。

  • TypeError - x 的数据类型既不是float16也不是float32。

  • ValueError - alpha 不等于1.0。

支持平台：

Ascend GPU CPU

样例：

>>> x = Tensor(np.array([-1, -2, 0, 2, 1]), mindspore.float32)
>>> elu = nn.ELU()
>>> result = elu(x)
>>> print(result)
[-0.63212055  -0.86466473  0.  2.  1.]