test=develop, fix nonzero astuple=true (#30647)

16427570 · 123malin · GitHub · 0eea5d71 · 16427570
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/tensor/search.py python/paddle/tensor/search.py +1 -1

未找到文件。
--- a/python/paddle/tensor/search.py
+++ b/python/paddle/tensor/search.py
@@ -399,7 +399,7 @@ def nonzero(x, as_tuple=False):
        for i in range(rank):
            list_out.append(
                layers.slice(
-                    outs, axes=[rank - 1], starts=[i], ends=[i + 1]))
+                    outs, axes=[1], starts=[i], ends=[i + 1]))
        return tuple(list_out)