remove redundant equal (PaddlePaddle#57986)

hitywt · Oct 24, 2023 · 165bf01 · 165bf01
1 parent f8cfed3
commit 165bf01
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/python/paddle/base/variable_index.py b/python/paddle/base/variable_index.py
@@ -344,14 +344,10 @@ def get_value_for_bool_tensor(var, item):
     empty_shape = [0] + list(var.shape[i:])
 
     def idx_not_empty(var, item):
-        from ..tensor import gather_nd
+        bool_2_idx = paddle.nonzero(item)
+        return paddle.gather_nd(var, bool_2_idx)
 
-        bool_2_idx = paddle.nonzero(item == True)
-        return gather_nd(var, bool_2_idx)
-
-    from paddle.static.nn import cond
-
-    return cond(
+    return paddle.static.nn.cond(
         item.any(),
         lambda: idx_not_empty(var, item),
         lambda: paddle.empty(empty_shape, var.dtype),