Tanhshrink
-
class
torch.nn.
Tanhshrink
[source] Applies the element-wise function:
- Shape:
Input: where * means, any number of additional dimensions
Output: , same shape as the input
Examples:
>>> m = nn.Tanhshrink() >>> input = torch.randn(2) >>> output = m(input)