From e30ccc5263d2e0a1cb9313538d79c2dfd62fb3be Mon Sep 17 00:00:00 2001 From: DesmonDay <908660116@qq.com> Date: Fri, 12 Jun 2020 20:06:43 +0800 Subject: [PATCH] modify temp variable name --- examples/SAGPool/layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/SAGPool/layers.py b/examples/SAGPool/layers.py index 650beef..f1f45fb 100644 --- a/examples/SAGPool/layers.py +++ b/examples/SAGPool/layers.py @@ -52,7 +52,7 @@ def topk_pool(gw, score, graph_id, ratio): max_num_nodes = L.cast(max_num_nodes, dtype="int32") index = L.arange(0, gw.num_nodes, dtype="int64") - temp = L.gather(graph_lod, graph_id, overwrite=False) + offset = L.gather(graph_lod, graph_id, overwrite=False) index = (index - temp) + (graph_id * max_num_nodes) index.stop_gradient = True -- GitLab