hardshrink_cn.rst

hardshrink

hardshrink 激活层。计算公式如下：

$$\begin{aligned} hardshrink(x)= \left\{ \begin{aligned} &x, & & if \ x > threshold \\\ &x, & & if \ x < -threshold \\\ &0, & & if \ others \end{aligned} \right. \end{aligned}$$

其中，x 为输入的 Tensor。

参数

• x (Tensor) - 输入的 Tensor，数据类型为：float32、float64。
• threshold (float，可选) - hard_shrink 激活计算公式中的 threshold 值。默认值为 0.5。
• name (str，可选) - 具体用法请参见 api_guide_Name，一般无需设置，默认值为 None。

返回

Tensor，数据类型和形状同 x 一致。

代码示例

COPY-FROM: paddle.nn.functional.hardshrink