diff --git a/pylops/torchoperator.py b/pylops/torchoperator.py index 0e3486f1..5e41f67f 100644 --- 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