Hardswish¶
- class torch.nn.Hardswish(inplace=False)[source]¶
逐元素应用 Hardswish 函数。
论文中描述的方法:Searching for MobileNetV3。
Hardswish 定义为
- 参数
inplace (布尔值) – 可选地就地执行操作。默认值:
False
- 形状
输入:,其中表示任意数量的维度。
输出:,与输入具有相同的形状。
示例
>>> m = nn.Hardswish() >>> input = torch.randn(2) >>> output = m(input)