提交 e30ccc52 编写于 作者: D DesmonDay

modify temp variable name

上级 a84c739b
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册