nnf_softshrink: Softshrink

View source: R/nnf-activation.R

nnf_softshrinkR Documentation

Softshrink

Description

Applies the soft shrinkage function elementwise

Usage

nnf_softshrink(input, lambd = 0.5)

Arguments

input

(N,*) tensor, where * means, any number of additional dimensions

lambd

the lambda (must be no less than zero) value for the Softshrink formulation. Default: 0.5


torch documentation built on May 29, 2024, 9:54 a.m.