Unlimited learning, half price | 50% off

Last chance! 50% off unlimited learning

Sale ends in


torch (version 0.14.2)

nn_hardswish: Hardswish module

Description

Applies the hardswish function, element-wise, as described in the paper: Searching for MobileNetV3

Usage

nn_hardswish()

Arguments

Shape

  • Input: (N,) where * means, any number of additional dimensions

  • Output: (N,), same shape as the input

Details

Hardswish(x)={0if x3,xif x+3,x(x+3)/6otherwise

Examples

Run this code
if (torch_is_installed()) {
if (FALSE) {
m <- nn_hardswish()
input <- torch_randn(2)
output <- m(input)
}

}

Run the code above in your browser using DataLab