chenxuanhong
29 lines
792 B
Diff
29 lines
792 B
Diff
--- /usr/local/lib/python3.5/dist-packages/torch/nn/modules/activation.py
|
|
+++ /usr/local/lib/python3.5/dist-packages/torch/nn/modules/activation.py
|
|
@@ -2,7 +2,7 @@
|
|
r"""Applies the element-wise function:
|
|
|
|
.. math::
|
|
- \text{Sigmoid}(x) = \sigma(x) = \frac{1}{1 + \exp(-x)}
|
|
+ \text{Sigmoid}(x) = \frac{1}{1 + \exp(-x)}
|
|
|
|
|
|
Shape:
|
|
@@ -10,7 +10,7 @@
|
|
dimensions
|
|
- Output: :math:`(N, *)`, same shape as the input
|
|
|
|
- .. image:: ../scripts/activation_images/Sigmoid.png
|
|
+ .. image:: scripts/activation_images/Sigmoid.png
|
|
|
|
Examples::
|
|
|
|
@@ -19,6 +19,7 @@
|
|
>>> output = m(input)
|
|
"""
|
|
|
|
- def forward(self, input: Tensor) -> Tensor:
|
|
+ @weak_script_method
|
|
+ def forward(self, input):
|
|
return torch.sigmoid(input)
|
|
|