diff --git a/functions.py b/functions.py index f19feeea6a0336a86ffe3bab04174799a5d99730..cf86c0b3d3ae5e62b507b97ee1b0b3e03b6a3f66 100644 --- a/functions.py +++ b/functions.py @@ -385,8 +385,9 @@ class ReLu(_Function): # TODO: Implement the derivative dx for this opetation and add the # result of the chain rule on self.dx. ####################################################################### - is_sup = self.x.data > 0 - self.dx = grad * is_sup + # is_sup = self.x.data > 0 + # self.dx = grad * is_sup + self.dx = grad * (self.x.data > 0) ####################################################################### # --------------------------- END OF YOUR CODE ------------------------ #######################################################################