Add Sigmoid

Author: lantiga

test/base.py

@@ -26,10 +26,11 @@ def base_test():
 
     model_1 = lambda x: F.softmax(F.elu(fc3(x)))
     model_2 = lambda x: F.softmax(F.tanh(fc3(x)))
+    model_3 = lambda x: F.softmax(F.sigmoid(fc3(x)))
 
     data = Variable(torch.rand(10,10))
 
-    out = model_0(data) + model_1(data) - model_2(data) + 1 - 2
+    out = model_0(data) + model_1(data) - model_2(data) + model_3(data) + 1 - 2
 
     out_path = 'out'
     if not os.path.isdir(out_path):

torch2c/emitters.py

@@ -452,6 +452,29 @@ def free_tpl(self):
 register(Tanh, torch.autograd._functions.pointwise.Tanh)
 
 
+class Sigmoid(Emitter):
+
+    def __init__(self, obj, prevfns):
+        Emitter.__init__(self, obj, prevfns)
+        self.def_vars({
+            'input': id(prevfns[0]),
+        })
+        self.infer_type_var = 'input'
+
+    def call_tpl(self):
+        return '''
+            TH${T}Tensor *$id = TH${T}Tensor_new();
+            THNN_${T}Sigmoid_updateOutput(NULL,$input,$id);
+            '''
+
+    def free_tpl(self):
+        return '''
+            TH${T}Tensor_free($id);
+            '''
+
+register(Sigmoid, torch.autograd._functions.pointwise.Sigmoid)
+
+
 class Noop(Emitter):
 
     def __init__(self, obj, prevfns):