nn_hardswish {torch} | R Documentation |
Hardswish module
Description
Applies the hardswish function, element-wise, as described in the paper: Searching for MobileNetV3
Usage
nn_hardswish()
Details
Shape
Input:
where
*
means, any number of additional dimensionsOutput:
, same shape as the input
Examples
if (torch_is_installed()) {
## Not run:
m <- nn_hardswish()
input <- torch_randn(2)
output <- m(input)
## End(Not run)
}
[Package torch version 0.13.0 Index]