add Substraction

functions.py

@@ -52,21 +52,21 @@ class Sub(_Function):
     def __init__(self, x, y):
         super().__init__("Sub", x, y)
         #######################################################################
-        # TODO: Implement the forward pass and put the result in self.result.
+        # Implement the forward pass and put the result in self.result.
         # The notbook provide you the formulas for this operation.
         #######################################################################
-        self.result = None
+        self.result = x.data - y.data
         #######################################################################
         # --------------------------- END OF YOUR CODE ------------------------
         #######################################################################
 
     def _backward(self, grad):
         #######################################################################
-        # TODO: Implement the derivative dx for this opetation and add the
+        # Implement the derivative dx for this opetation and add the
         # result of the chain rule on self.dx.
         #######################################################################
-        self.dx = None
-        self.dy = None
+        self.dx = grad
+        self.dy = -grad
         #######################################################################
         # --------------------------- END OF YOUR CODE ------------------------
         #######################################################################

variable.py

@@ -169,9 +169,9 @@ class Variable:
                 children = []
             if not len(children):
                 #######################################################################
-                # TODO: Call the backward of the operation that has build this Variable
+                # Call the backward of the operation that has build this Variable
                 #######################################################################
-                pass
+                self.grad_fn.backward(self.grad, retain_graph)
                 #######################################################################
                 # --------------------------- END OF YOUR CODE ------------------------
                 #######################################################################