From 370b3f3665d22751a2af047ea0dc7f114af6d042 Mon Sep 17 00:00:00 2001
From: 123malin <malin10@baidu.com>
Date: Mon, 1 Feb 2021 17:02:50 +0800
Subject: [PATCH] test=develop, fix nonzero astuple=true (#30647) (#30806)

---
 python/paddle/tensor/search.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/tensor/search.py b/python/paddle/tensor/search.py
index 32f7bf373bb..95f8fa449bd 100644
--- 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)
 
 
-- 
GitLab