Added batch as parameter

rohanbabbar04 · rohanbabbar04 · commit 69eb2046b141 · 2023-02-25T09:30:55.000+05:30
diff --git a/pylops/torchoperator.py b/pylops/torchoperator.py
@@ -78,8 +78,7 @@ def _register_torchop(self, batch: bool):
         # choose _matvec and _rmatvec
         self._hmatvec: Callable
         self._hrmatvec: Callable
-
-        if batch:
+        if not batch:
             self._hmatvec = lambda x: self.Op @ x
             self._hrmatvec = lambda x: self.Op.H @ x
         else:
