diff --git a/functions.py b/functions.py index cf86c0b3d3ae5e62b507b97ee1b0b3e03b6a3f66..604b225f962b8a8c60a88f4a1188ebade0641bd7 100644 --- a/functions.py +++ b/functions.py @@ -387,7 +387,7 @@ class ReLu(_Function): ####################################################################### # is_sup = self.x.data > 0 # self.dx = grad * is_sup - self.dx = grad * (self.x.data > 0) + self.dx = (self.x.data > 0) * grad ####################################################################### # --------------------------- END OF YOUR CODE ------------------------ #######################################################################