Rate this Page

Tanhshrink#

classtorch.nn.modules.activation.Tanhshrink(*args,**kwargs)[source]#

Applies the element-wise Tanhshrink function.

Tanhshrink(x)=xtanh(x)\text{Tanhshrink}(x) = x - \tanh(x)
Shape:
  • Input:()(*), where* means any number of dimensions.

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

../_images/Tanhshrink.png

Examples:

>>>m=nn.Tanhshrink()>>>input=torch.randn(2)>>>output=m(input)
forward(input)[source]#

Runs the forward pass.

Return type

Tensor