2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.mint.nn.functional.elu.rst 998 Bytes
一键复制 编辑 原始数据 按行查看 历史

mindspore.mint.nn.functional.elu

.. py:function:: mindspore.mint.nn.functional.elu(input, alpha=1.0)

    指数线性单元激活函数。

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

    .. math::

        \text{ELU}(x)= \left\{
        \begin{array}{align}
            \alpha(e^{x}  - 1) & \text{if } x \le 0\\
            x & \text{if } x \gt 0\\
        \end{array}\right.

    其中, :math:`x` 表示输入Tensor `input` , :math:`\alpha` 表示 `alpha` 参数, `alpha` 决定ELU的平滑度。

    ELU函数图:

    .. image:: ../images/ELU.png
        :align: center

    参数:
        - **input** (Tensor) - ELU的输入,为任意维度的Tensor。
        - **alpha** (float, 可选) - ELU的alpha值,数据类型为float。默认值: ``1.0`` 。

    返回:
        Tensor,输出的shape和数据类型与 `input` 相同。

    异常:
        - **TypeError** - 如果 `alpha` 不是float。


马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助