Merge pull request #1076 from Trusted-AI/development_issue_1073

Add missing to device transfer

Author: beat-buesser

art/attacks/evasion/projected_gradient_descent/projected_gradient_descent_pytorch.py

@@ -293,7 +293,7 @@ def _compute_perturbation(  # pylint: disable=W0221
 
         # Apply mask
         if mask is not None:
-            grad = torch.where(mask == 0.0, torch.tensor(0.0), grad)
+            grad = torch.where(mask == 0.0, torch.tensor(0.0).to(self.estimator.device), grad)
 
         # Apply norm bound
         if self.norm in ["inf", np.inf]: