From 9209f341fc0ff9e79f5f52320955c5aba358ed09 Mon Sep 17 00:00:00 2001
From: AshburnLee <1578034415@qq.com>
Date: Wed, 14 Oct 2020 10:40:15 +0800
Subject: [PATCH] Made a change in python API to support the GPU version of
 unique OP working in static graph (#27899)

* Made a change to support unique op working in static graph

* Correct python format error
---
 python/paddle/tensor/manipulation.py | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py
index 73a37253828..ba7a1f1ef23 100644
--- a/python/paddle/tensor/manipulation.py
+++ b/python/paddle/tensor/manipulation.py
@@ -669,21 +669,24 @@ def unique(x,
     }
     out = helper.create_variable_for_type_inference(
         dtype=x.dtype, stop_gradient=True)
+    indices = helper.create_variable_for_type_inference(
+        dtype=attr_dtype, stop_gradient=True)
     inverse = helper.create_variable_for_type_inference(
         dtype=attr_dtype, stop_gradient=True)
-    outputs = {"Out": out, "Index": inverse}
+    counts = helper.create_variable_for_type_inference(
+        dtype=attr_dtype, stop_gradient=True)
+    outputs = {
+        "Out": out,
+        "Indices": indices,
+        "Index": inverse,
+        "Counts": counts
+    }
     outs = [out]
     if return_index:
-        indices = helper.create_variable_for_type_inference(
-            dtype=attr_dtype, stop_gradient=True)
-        outputs["Indices"] = indices
         outs.append(indices)
     if return_inverse:
         outs.append(inverse)
     if return_counts:
-        counts = helper.create_variable_for_type_inference(
-            dtype=attr_dtype, stop_gradient=True)
-        outputs["Counts"] = counts
         outs.append(counts)
 
     helper.append_op(
-- 
GitLab