- class torch.nn.Hardswish(inplace=False)¶
Applies the Hardswish function, element-wise, as described in the paper: Searching for MobileNetV3.
Hardswish is defined as:
inplace (bool) – can optionally do the operation in-place. Default:
Input: , where means any number of dimensions.
Output: , same shape as the input.
>>> m = nn.Hardswish() >>> input = torch.randn(2) >>> output = m(input)