diff --git a/fluid/adversarial/advbox/attacks/saliency.py b/fluid/adversarial/advbox/attacks/saliency.py
index 7922192fc92131051c4edc85af200b887b3e7b7e..4372025251bbfa280ec83eca9ab4201b65076f3d 100644
--- a/fluid/adversarial/advbox/attacks/saliency.py
+++ b/fluid/adversarial/advbox/attacks/saliency.py
@@ -57,7 +57,7 @@ class SaliencyMapAttack(Attack):
                     return adversary
 
                 # stop if mask is all zero
-                if self._is_zero_mask(mask):
+                if not any(mask):
                     return adversary
 
                 # get pixel location with highest influence on class
@@ -82,21 +82,6 @@ class SaliencyMapAttack(Attack):
 
         return adversary
 
-    def _is_zero_mask(self, mask):
-        """
-        The elements in mask are all zero or not.
-        Args:
-            mask(list): Each modified pixel with border value is set to zero in mask.
-        Return: bool
-        """
-        is_all_zero = True
-        for item in mask:
-            if item != 0:
-                is_all_zero = False
-                break
-
-        return is_all_zero
-
     def _saliency_map(self, image, target, labels, mask, fast=False):
         """
         Get pixel location with highest influence on class.