softshrink_cn.rst 1.1 KB
Newer Older
S
swtkiwi 已提交
1 2 3 4
.. _cn_api_nn_cn_softshrink:

softshrink
-------------------------------
S
swtkiwi 已提交
5

6
.. py:function:: paddle.nn.functional.softshrink(x, threshold=0.5, name=None)
S
swtkiwi 已提交
7

8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
softshrink激活层

.. math::

    softshrink(x)= \begin{cases}
                    x - threshold, \text{if } x > threshold \\
                    x + threshold, \text{if } x < -threshold \\
                    0,  \text{otherwise}
                    \end{cases}

其中,:math:`x` 为输入的 Tensor

参数:
::::::::::
 - x (Tensor) - 输入的 ``Tensor`` ,数据类型为:float32、float64。
 - threshold (float, 可选) - softshrink激活计算公式中的threshold值,必须大于等于零。默认值为0.5。
 - name (str, 可选) - 操作的名称(可选,默认值为None)。更多信息请参见 :ref:`api_guide_Name`。

返回
::::::::::
    ``Tensor`` ,数据类型和形状同 ``x`` 一致。

代码示例
::::::::::

.. code-block:: python

    import paddle
    import paddle.nn.functional as F
    import numpy as np

    paddle.disable_static()

    x = paddle.to_tensor(np.array([-0.9, -0.2, 0.1, 0.8]))
    out = F.softshrink(x) # [-0.4, 0, 0, 0.3]