硬饱和度 ¶
-
class torch.nn.Hardsigmoid(inplace=False)[source][source]
应用元素级的 Hardsigmoid 函数。
Hardsigmoid 定义如下:
Hardsigmoid(x)=⎩⎨⎧01x/6+1/2if x≤−3,if x≥+3,otherwise
- 参数:
inplace(布尔值)- 可选地进行原地操作。默认: False
- 形状:
-
示例:
>>> m = nn.Hardsigmoid()
>>> input = torch.randn(2)
>>> output = m(input)