minor: added __call__ to TorchOperator

PyLops · Feb 27, 2023 · fee227f · fee227f
1 parent 4428dda
commit fee227f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pylops/torchoperator.py b/pylops/torchoperator.py
@@ -82,6 +82,9 @@ def __init__(
             )
         self.Top = _TorchOperator.apply
 
+    def __call__(self, x):
+        return self.apply(x)
+
     def apply(self, x: TensorTypeLike) -> TensorTypeLike:
         """Apply forward pass to input vector