nn_softsign: Softsign module

nn_softsignR Documentation

Softsign module

Description

Applies the element-wise function:

\mbox{SoftSign}(x) = \frac{x}{ 1 + |x|}

Usage

nn_softsign()

Shape

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

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

Examples

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

torch documentation built on June 7, 2023, 6:19 p.m.