fix: fixed np.sum in art/defences/postprocessor/gaussian_noise.py

Signed-off-by: GianlucaMega <[email protected]>

Author: g.mega

art/defences/postprocessor/gaussian_noise.py

@@ -69,7 +69,7 @@ def __call__(self, preds: np.ndarray) -> np.ndarray:
             # Finally normalize probability output
             if all_probability:
                 post_preds[post_preds < 0.0] = 0.0
-                sums = np.sum(post_preds, axis=1)
+                sums = np.sum(post_preds, axis=1, keepdims=True)
                 post_preds /= sums
         else:
             post_preds[post_preds < 0.0] = 0.0