mindspore2022/docs/api/api_python/mindspore.ops/mindspore.nn.elu.rst

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.]