硬件加速 ¶

class torch.nn.Hardswish(inplace=False)

逐元素应用 Hardswish 函数。

方法见论文：寻找 MobileNetV3。

Hardswish 定义为：

$$\text{Hardswish}(x) = \begin{cases} 0 & \text{if~} x \le -3, \\ x & \text{if~} x \ge +3, \\ x \cdot (x + 3) /6 & \text{otherwise} \end{cases}$$

参数:

inplace（布尔值）- 可选地进行原地操作。默认： False

形状：

• 输入： $(*)$ ，其中 $*$ 表示任意数量的维度。

• 输出： $(*)$ ，与输入具有相同的形状。

示例：

>>> m = nn.Hardswish()
>>> input = torch.randn(2)
>>> output = m(input)