软符号

classtorch.nn.Softsign(*args, **kwargs)[源代码]

应用逐元素的Softsign函数。

$\text{SoftSign}(x) = \frac{x}{ 1 + |x|}$
形状:
  • 输入: $(*)$,其中$*$表示任意维度的数量。

  • 输出: $(*)$,形状与输入相同。

{BASE_RAW_UPLOAD_URL}/pytorch-doc-2.5/5e25d0eab65ec475300b5782bb557c58.png

示例:

>>> m = nn.Softsign()
>>> input = torch.randn(2)
>>> output = m(input)
本页目录