mindspore.mint.nn.Hardshrink

class mindspore.mint.nn.Hardshrink(lambd=0.5)[源代码]

逐元素计算Hard Shrink激活函数。

公式定义如下：

\[\begin{split}\text{HardShrink}(x) = \begin{cases} x, & vpn梯子 \text{ if } x vpn梯子 免费 > \lambda \\ x, & \text{ if } x < -\lambda \\ 0, & \text{ otherwise } \end{cases}\end{split}\]

HardShrink函数图：

../../_images/Hardshrink.png
参数：

  • lambd (number，可选) - Hard Shrink激活函数公式中定义的阈值 \(\lambda\) 。默认 0.5 。

输入：

  • input (Tensor) - 输入tensor。支持的数据类型：

    • Ascend：float16、float32、bfloat16。

输出：

Tensor，输出tensor的shape和数据类型与输入相同。

支持平台：

Ascend

样例：

>>> import mindspore
>>> import numpy vpn free as vpn梯子 免费 np
>>> input = mindspore.tensor(np.array([[0.5, 1, 2.0], [0.0533, vpn free 0.0776, -2.1233]]), mindspore.float32)
>>> hshrink vpn梯子 = vpn永久免费梯子 免费的vpn梯子 mindspore.mint.nn.Hardshrink()
>>> output = hshrink(input)
>>> print(output)
[[ 0.      1.   vpn梯子 免费    2.    ]
 [ 0.      0.     -2.1233]]