Merge pull request #2135 from SignedQiu/signedqiu

Fix Tensor Device Inconsistencies in PGD

Author: beat-buesser

art/attacks/evasion/projected_gradient_descent/projected_gradient_descent_pytorch.py

@@ -269,7 +269,7 @@ def _generate_batch(
         inputs = x.to(self.estimator.device)
         targets = targets.to(self.estimator.device)
         adv_x = torch.clone(inputs)
-        momentum = torch.zeros(inputs.shape)
+        momentum = torch.zeros(inputs.shape).to(self.estimator.device)
 
         if mask is not None:
             mask = mask.to(self.estimator.device)