Softshrink¶ class torch.nn.Softshrink(lambd=0.5)[source]¶ 按元素应用软阈值函数。 SoftShrinkage(x)={x−λ, if x>λx+λ, if x<−λ0, otherwise \text{SoftShrinkage}(x) = \begin{cases} x - \lambda, & \text{ if } x > \lambda \\ x + \lambda, & \text{ if } x < -\lambda \\ 0, & \text{ otherwise } \end{cases} SoftShrinkage(x)=⎩⎨⎧x−λ,x+λ,0, if x>λ if x<−λ otherwise 参数 lambd (float) – Softshrink 公式的 λ\lambdaλ 值(必须不小于零)。默认值:0.5 形状 输入:(∗)(*)(∗),其中 ∗*∗ 表示任意数量的维度。 输出: (∗)(*)(∗), 与输入形状相同。 示例 >>> m = nn.Softshrink() >>> input = torch.randn(2) >>> output = m(input)