2.6K Star 8.6K Fork 4.9K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
mindspore.ops.func_elu.rst 1.25 KB
一键复制 编辑 原始数据 按行查看 历史
俞涵 提交于 2024-02-27 12:05 . [i] modify format

mindspore.ops.elu

.. py:function:: mindspore.ops.elu(input_x, 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_x` , :math:`\alpha` 表示 `alpha` 参数, `alpha` 决定ELU的平滑度。
    ELU相关图参见 `ELU <https://en.wikipedia.org/wiki/Activation_function#/media/File:Activation_elu.svg>`_  。

    ELU函数图:

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

    参数:
        - **input_x** (Tensor) - ELU的输入,为任意维度的Tensor,数据类型为float16或float32。
        - **alpha** (float, 可选) - ELU的alpha值,数据类型为float,目前仅支持1.0。默认值: ``1.0`` 。

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

    异常:
        - **TypeError** - 如果 `alpha` 不是float。
        - **TypeError** - 如果 `input_x` 的数据类型既不是float16,也不是float32。
        - **ValueError** - 如果 `alpha` 不等于1.0。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.q1

搜索帮助

0d507c66 1850385 C8b1a773 1850385