Added operation with gradients in modules

JVGD · Oct 23, 2021 · 524a492 · 524a492
1 parent 3f39c18
commit 524a492
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/experiment1/modules.py b/experiment1/modules.py
@@ -8,8 +8,7 @@ def __init__(self) -> None:
 
     def __call__(self, X: T.Tensor) -> T.Tensor:
         """Calling the adding operation without gradient"""
-        with T.no_grad():
-            return X[0] + X[1]
+        return X[0] + X[1]
 
 
 class Substracter(object):
@@ -19,8 +18,7 @@ def __init__(self) -> None:
 
     def __call__(self, X: T.Tensor) -> T.Tensor:
         """Calling the adding operation without gradient"""
-        with T.no_grad():
-            return X[0] - X[1]
+        return X[0] - X[1]
 
 
 if __name__ == "__main__":