patch v2 relu

functions.py

@@ -385,7 +385,7 @@ class ReLu(_Function):
         # TODO: Implement the derivative dx for this opetation and add the
         # result of the chain rule on self.dx.
         #######################################################################
-        self.dx = grad * 0 if self.x.data <= 0 else 1
+        self.dx = grad * (0 if self.x.data <= 0 else 1)
         #######################################################################
         # --------------------------- END OF YOUR CODE ------------------------
         #######################################################################