nn_hardshrink function

Hardshwink module

Applies the hard shrinkage function element-wise:

nn_hardshrink(lambd = 0.5)

Arguments

• lambd: the $\lambda$ value for the Hardshrink formulation. Default: 0.5

Details

Shape

• Input: $(N, *)$ where * means, any number of additional dimensions
• Output: $(N, *)$, same shape as the input

Examples

if (torch_is_installed()) {
m <- nn_hardshrink()
input <- torch_randn(2)
output <- m(input)
}