Shortcuts

Sigmoid

class torch.nn.Sigmoid(*args, **kwargs)[源代码]

逐元素应用Sigmoid函数。

Sigmoid(x)=σ(x)=11+exp(x)\text{Sigmoid}(x) = \sigma(x) = \frac{1}{1 + \exp(-x)}
Shape:
  • 输入:()(*),其中 * 表示任意数量的维度。

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

../_images/Sigmoid.png

示例:

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