change nn.sigmoid

2020-05-07 15:56:43 +08:00 · 2020-05-07 15:56:43 +08:00 · 503da59a3b
parent d2b571e2bc
commit 503da59a3b
1 changed files with 1 additions and 1 deletions
--- a/python/jittor/nn.py
+++ b/python/jittor/nn.py
@ -633,7 +633,7 @@ class Sigmoid(Module):
    def __init__(self):
        super().__init__()
    def execute(self, x) :
-        return 1 / (1 + jt.exp(-x))
+        return x.sigmoid()

 def resize(x, size, mode="nearest"):
    img = x