2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.mint.nn.functional.softplus.rst 1.10 KB
一键复制 编辑 原始数据 按行查看 历史
俞涵 提交于 2024-06-24 11:36 . modify format

mindspore.mint.nn.functional.softplus

.. py:function:: mindspore.mint.nn.functional.softplus(input, beta=1, threshold=20)

    将softplus函数作用于 `input` 的每个元素上。

    softplus函数如下所示,其中x是 `input` 中的元素:

    .. math::

        \text{output} = \frac{1}{beta}\log(1 + \exp(\text{beta * x}))

    当 :math:`input * beta > threshold` 时,为保证数值稳定性,softplus的实现被转换为线性函数。

    参数:
        - **input** (Tensor) - 任意维度的输入Tensor。支持数据类型:

          - Ascend:float16、float32、bfloat16。

        - **beta** (number.Number,可选) - softplus函数中的缩放参数。默认值:``1`` 。
        - **threshold** (number.Number,可选) - 为了数值稳定性,softplus函数转换为线性函数的阈值参数。默认值:``20`` 。

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

    异常:
        - **TypeError** - `input` 不是Tensor。
        - **TypeError** - `input` 的数据类型非float16、float32、bfloat16。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.1

搜索帮助