PyTorch
torch / nn / torch.nn
Hardshrink¶
-
class
torch.nn.
Hardshrink
(lambd: float = 0.5)[source]¶ Applies the hard shrinkage function element-wise:
HardShrink(x)=⎩⎪⎪⎨⎪⎪⎧x,x,0, if x>λ if x<−λ otherwise - Parameters
lambd – the λ value for the Hardshrink formulation. Default: 0.5
- Shape:
Input: (N,∗) where * means, any number of additional dimensions
Output: (N,∗) , same shape as the input
Examples:
>>> m = nn.Hardshrink() >>> input = torch.randn(2) >>> output = m(input)
此页内容是否对您有帮助
感谢反馈!