未验证 提交 e3b4f049 编写于 作者: W wawltor 提交者: GitHub

Bug fix: fix a small bug

......@@ -53,7 +53,7 @@ def topk_pool(gw, score, graph_id, ratio):
index = L.arange(0, gw.num_nodes, dtype="int64")
offset = L.gather(graph_lod, graph_id, overwrite=False)
index = (index - temp) + (graph_id * max_num_nodes)
index = (index - offset) + (graph_id * max_num_nodes)
index.stop_gradient = True
# padding
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册