PyTorch
torch / nn / torch.nn
Hardsigmoid¶
-
class
torch.nn.
Hardsigmoid
(inplace: bool = False)[source]¶ Applies the element-wise function:
Hardsigmoid(x)=⎩⎪⎪⎨⎪⎪⎧01x/6+1/2if x≤−3,if x≥+3,otherwise- Parameters
inplace – can optionally do the operation in-place. Default:
False
- Shape:
Input: (N,∗) where * means, any number of additional dimensions
Output: (N,∗) , same shape as the input
Examples:
>>> m = nn.Hardsigmoid() >>> input = torch.randn(2) >>> output = m(input)
此页内容是否对您有帮助
感谢反馈!