• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1mindspore.mint.nn.functional.elu
2===================================
3
4.. py:function:: mindspore.mint.nn.functional.elu(input, alpha=1.0)
5
6    指数线性单元激活函数。
7
8    对输入的每个元素计算ELU。该激活函数定义如下：
9
10    .. math::
11
12        \text{ELU}(x)= \left\{
13        \begin{array}{align}
14            \alpha(e^{x}  - 1) & \text{if } x \le 0\\
15            x & \text{if } x \gt 0\\
16        \end{array}\right.
17
18    其中， :math:`x` 表示输入Tensor `input` ， :math:`\alpha` 表示 `alpha` 参数， `alpha` 决定ELU的平滑度。
19
20    ELU函数图：
21
22    .. image:: ../images/ELU.png
23        :align: center
24
25    参数：
26        - **input** (Tensor) - ELU的输入，为任意维度的Tensor。
27        - **alpha** (float, 可选) - ELU的alpha值，数据类型为float。默认值： ``1.0`` 。
28
29    返回：
30        Tensor，输出的shape和数据类型与 `input` 相同。
31
32    异常：
33        - **TypeError** - 如果 `alpha` 不是float。
34
35
36