Merge babb4ad into 391e2c0

chainer · Aug 14, 2017 · c1292ca · c1292ca
2 parents 391e2c0 + babb4ad
commit c1292ca
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/chainer/functions/math/basic_math.py b/chainer/functions/math/basic_math.py
@@ -47,7 +47,7 @@ def _preprocess_const(x, value):
     return utils.force_type(x.dtype, value)
 
 
-class Neg(function.Function):
+class Neg(function_node.FunctionNode):
 
     @property
     def label(self):
@@ -60,8 +60,8 @@ def forward(self, x):
         self.retain_inputs(())
         return utils.force_array(-x[0]),
 
-    def backward(self, x, gy):
-        return utils.force_array(-gy[0]),
+    def backward(self, indexes, gy):
+        return -gy[0],
 
 
 def neg(self):  # -x
@@ -70,7 +70,7 @@ def neg(self):  # -x
     Returns:
         ~chainer.Variable: Output variable.
     """
-    return Neg()(self)
+    return Neg().apply((self,))[0]
 
 
 class Absolute(function.Function):