Tensor Attributes

Tensor Views

Complex Numbers

Quantization

Distributed RPC Framework

Type Info

Named Tensors

Named Tensors operator coverage

API

torch / nn / torch.nn

Tanhshrink¶

class torch.nn.Tanhshrink[source]¶

Applies the element-wise function:

\text{Tanhshrink}(x) = x - \tanh(x)

Shape:

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

../_images/Tanhshrink.png

Examples:

>>> m = nn.Tanhshrink()
>>> input = torch.randn(2)
>>> output = m(input)

此页内容是否对您有帮助