Tensor Attributes

Tensor Views

Complex Numbers

Quantization

Distributed RPC Framework

Type Info

Named Tensors

Named Tensors operator coverage

API

torch / nn / torch.nn

Sigmoid¶

class torch.nn.Sigmoid[source]¶

Applies the element-wise function:

\text{Sigmoid}(x) = \sigma(x) = \frac{1}{1 + \exp(-x)}

Shape:

Input: $(N, *)$ where * means, any number of additional dimensions
Output: $(N, *)$ , same shape as the input

../_images/Sigmoid.png

Examples:

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

此页内容是否对您有帮助