Skip to content
Snippets Groups Projects
Commit b82bc6c5 authored by Maxxhim's avatar Maxxhim
Browse files

patch v4 relu

parent 1a5505ae
No related branches found
No related tags found
No related merge requests found
...@@ -385,7 +385,7 @@ class ReLu(_Function): ...@@ -385,7 +385,7 @@ class ReLu(_Function):
# TODO: Implement the derivative dx for this opetation and add the # TODO: Implement the derivative dx for this opetation and add the
# result of the chain rule on self.dx. # result of the chain rule on self.dx.
####################################################################### #######################################################################
is_sup = 0 if self.x.data <= 0 else 1 is_sup = self.x.data > 0
self.dx = grad * is_sup self.dx = grad * is_sup
####################################################################### #######################################################################
# --------------------------- END OF YOUR CODE ------------------------ # --------------------------- END OF YOUR CODE ------------------------
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment