Merge pull request #1785 from abigailgold/dev_1.11.1_fix_pytorch

Fix exception in pytorch classifier predict method

Author: beat-buesser

art/estimators/classification/pytorch.py

@@ -322,7 +322,7 @@ def predict(  # pylint: disable=W0221
             output = model_outputs[-1]
             output = output.detach().cpu().numpy().astype(np.float32)
             if len(output.shape) == 1:
-                output = np.expand_dims(output.detach().cpu().numpy(), axis=1).astype(np.float32)
+                output = np.expand_dims(output, axis=1).astype(np.float32)
 
             results_list.append(output)