fix the problem of slice infer shape (#42568) (#43246)

PaddlePaddle · Jun 7, 2022 · f1b4e4d · f1b4e4d
1 parent e09803c
commit f1b4e4d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/paddle/fluid/variable_index.py b/python/paddle/fluid/variable_index.py
@@ -306,8 +306,9 @@ def idx_empty(var):
         return paddle.empty(var_shape, dtype=var.dtype)
 
     from .layers.control_flow import cond
-    return cond(item.any(), lambda: idx_not_empty(var, item),
-                lambda: idx_empty(var))
+    return cond(
+        paddle.logical_not(item.any()), lambda: idx_empty(var),
+        lambda: idx_not_empty(var, item))
 
 
 def _getitem_impl_(var, item):