Skip to content

P
PGL

PaddlePaddle / PGL

通知 76
Star 4
Fork 0
P
PGL
未验证 提交 e3b4f049 编写于 作者: W wawltor 提交者： GitHub
浏览文件
操作

Bug fix: fix a small bug

上级 d882cc6b b48a99fd
隐藏空白更改
内联 并排
Showing with 1 addition and 1 deletion
examples/SAGPool/layers.py
浏览文件 @ e3b4f049
...@@ -53,7 +53,7 @@ def topk_pool(gw, score, graph_id, ratio): ...@@ -53,7 +53,7 @@ def topk_pool(gw, score, graph_id, ratio):
index = L.arange(0, gw.num_nodes, dtype="int64") index = L.arange(0, gw.num_nodes, dtype="int64")
offset = 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 = (index - offset) + (graph_id * max_num_nodes)
index.stop_gradient = True index.stop_gradient = True
# padding # padding
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或