pytorch / 1.8.0 / generated / torch.nn.sigmoid.html /

Sigmoid

class torch.nn.Sigmoid [source]

Applies the element-wise function:

Sigmoid ( x ) = σ ( x ) = 1 1 + exp ( x ) \text{Sigmoid}(x) = \sigma(x) = \frac{1}{1 + \exp(-x)}
Shape:
  • Input: ( N , ) (N, *) where * means, any number of additional dimensions
  • Output: ( N , ) (N, *) , same shape as the input
../_images/Sigmoid.png

Examples:

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

© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.nn.Sigmoid.html