評價此頁

Tanhshrink#

class torch.nn.Tanhshrink(*args, **kwargs)[source]#

逐元素應用 Tanhshrink 函式。

Tanhshrink(x)=xtanh(x)\text{Tanhshrink}(x) = x - \tanh(x)
形狀
  • 輸入: ()(*),其中 * 表示任意數量的維度。

  • 輸出: ()(*),形狀與輸入相同。

../_images/Tanhshrink.png

示例

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

執行前向傳播。

返回型別

張量