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函数图:
- 参数:
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
样例:
>>> 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.]