Bugfix: return float for relu

dantp-ai · Mar 22, 2024 · b35d173 · b35d173
1 parent 1b9cfe1
commit b35d173
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/minitorch/operators.py b/minitorch/operators.py
@@ -83,7 +83,7 @@ def relu(x: float) -> float:
     (See https://en.wikipedia.org/wiki/Rectifier_(neural_networks) .)
     """
 
-    return x if x > 0.0 else 0
+    return x if x > 0.0 else 0.0
 
 
 EPS = 1e-6