Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PGL
提交
46dd55da
P
PGL
项目概览
PaddlePaddle
/
PGL
通知
76
Star
4
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
11
列表
看板
标记
里程碑
合并请求
1
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PGL
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
11
Issue
11
列表
看板
标记
里程碑
合并请求
1
合并请求
1
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
46dd55da
编写于
4月 05, 2020
作者:
K
kirayummy
提交者:
GitHub
4月 05, 2020
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #37 from Yelrose/master
Add Graph Normalization Layers
上级
2a8223af
50ab64bb
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
26 addition
and
1 deletion
+26
-1
pgl/layers/graph_pool.py
pgl/layers/graph_pool.py
+26
-1
未找到文件。
pgl/layers/graph_pool.py
浏览文件 @
46dd55da
...
...
@@ -19,7 +19,7 @@ from pgl import graph_wrapper
from
pgl.utils
import
paddle_helper
from
pgl.utils
import
op
__all__
=
[
'graph_pooling'
]
__all__
=
[
'graph_pooling'
,
'graph_norm'
]
def
graph_pooling
(
gw
,
node_feat
,
pool_type
):
...
...
@@ -40,3 +40,28 @@ def graph_pooling(gw, node_feat, pool_type):
graph_feat
=
op
.
nested_lod_reset
(
node_feat
,
gw
.
graph_lod
)
graph_feat
=
fluid
.
layers
.
sequence_pool
(
graph_feat
,
pool_type
)
return
graph_feat
def
graph_norm
(
gw
,
feature
):
"""Implementation of graph normalization
Reference Paper: BENCHMARKING GRAPH NEURAL NETWORKS
Each node features is divied by sqrt(num_nodes) per graphs.
Args:
gw: Graph wrapper object (:code:`StaticGraphWrapper` or :code:`GraphWrapper`)
feature: A tensor with shape (num_nodes, hidden_size)
Return:
A tensor with shape (num_nodes, hidden_size)
"""
nodes
=
fluid
.
layers
.
fill_constant
(
[
gw
.
num_nodes
,
1
],
dtype
=
"float32"
,
value
=
1.0
)
norm
=
graph_pooling
(
gw
,
nodes
,
pool_type
=
"sum"
)
norm
=
fluid
.
layers
.
sqrt
(
norm
)
feature_lod
=
op
.
nested_lod_reset
(
feature
,
gw
.
graph_lod
)
norm
=
fluid
.
layers
.
sequence_expand_as
(
norm
,
feature_lod
)
norm
.
stop_gradient
=
True
return
feature_lod
/
norm
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录