Hardswish¶
-
class
torch.nn.
Hardswish
(inplace=False)[source]¶ Applies the hardswish function, element-wise, as described in the paper:
- Parameters
inplace – can optionally do the operation in-place. Default:
False
- Shape:
Input: , where means any number of dimensions.
Output: , same shape as the input.
Examples:
>>> m = nn.Hardswish() >>> input = torch.randn(2) >>> output = m(input)