Merge pull request #11 from mrava87/master

Remove .to for numpy arrays

Author: mrava87

pylops_gpu/TorchOperator.py

@@ -40,7 +40,7 @@ def forward(ctx, x, forw, adj, pylops, device):
         if ctx.pylops:
             if ctx.device == 'cpu':
                 # move y to torch and device
-                y = torch.from_numpy(y).to(ctx.device)
+                y = torch.from_numpy(y)
             else:
                 # move y to torch and device
                 y = from_dlpack(y.toDlpack())
@@ -62,10 +62,10 @@ def backward(ctx, y):
         # prepare output
         if ctx.pylops:
             if ctx.device == 'cpu':
-                x = torch.from_numpy(x).to(ctx.device)
+                x = torch.from_numpy(x)
             else:
                 x = from_dlpack(x.toDlpack())
-        return  x, None, None, None, None
+        return x, None, None, None, None
 
 
 class TorchOperator():