mindspore.nn.ELU

class mindspore.nn.ELU(alpha=1.0)

指数线性单元激活函数(Exponential Linear Unit activation function)。

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

$$E_i=\{\begin{array}{ll}{x}_i,& \text{if }{x}_i\ge 0;\\ \alpha \ast (\exp (x_i)-1),& \text{otherwise.}\end{array}$$

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

ELU函数图：

参数：

• alpha (float) - ELU的alpha值，数据类型为浮点数，目前仅支持1.0。默认值： 1.0 。

输入：

• input_x (Tensor) - 用于计算ELU的任意维度的Tensor，数据类型为float16或float32。

输出：

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

异常：

• TypeError - alpha 不是浮点数。

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

• ValueError - alpha 不等于1.0。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> from mindspore import Tensor, nn
>>> import numpy as np
>>> 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.]