|
- mindspore.nn.ELU
- =================
-
- .. py:class:: mindspore.nn.ELU(alpha=1.0)
-
- 指数线性单元激活函数（Exponential Linear Uint activation function）。
-
- 对输入的每个元素计算ELU。该激活函数定义如下：
-
- .. math::
- E_{i} =
- \begin{cases}
- x, &\text{if } x \geq 0; \cr
- \text{alpha} * (\exp(x_i) - 1), &\text{otherwise.}
- \end{cases}
-
-
- ELU相关图参见 `ELU <https://en.wikipedia.org/wiki/Activation_function#/media/File:Activation_elu.svg>`_ 。
-
- **参数** ：
-
- - **alpha** (`float`) – ELU的alpha值，数据类型为浮点数。默认值：1.0。
-
- **输入** ：
-
- - **x** （Tensor） - 用于计算ELU的Tensor，数据类型为float16或float32。shape为 :math:`(N,*)` ，:math:`*` 表示任意的附加维度数。
-
- **输出** ：
-
- Tensor，具有与 `x` 相同的数据类型和shape。
-
- **异常** ：
-
- - **TypeError：** `alpha` 不是浮点数。
- - **TypeError：** `x` 的数据类型既不是float16也不是float32。
- - **ValueError：** `alpha` 不等于1.0。
-
- **支持平台** ：
-
- `Ascend` `GPU` `CPU`
-
- **样例** :
-
- .. code-block::
-
- >>> 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.]
|
