From 50ab64bb720be4b4692936e619eac4f517fa5093 Mon Sep 17 00:00:00 2001
From: Yelrose <270018958@qq.com>
Date: Fri, 3 Apr 2020 17:02:26 +0800
Subject: [PATCH] add graph_norm

---
 pgl/layers/graph_pool.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/pgl/layers/graph_pool.py b/pgl/layers/graph_pool.py
index a1b6249..fbe6c50 100644
--- a/pgl/layers/graph_pool.py
+++ b/pgl/layers/graph_pool.py
@@ -52,11 +52,10 @@ def graph_norm(gw, feature):
     Args:
         gw: Graph wrapper object (:code:`StaticGraphWrapper` or :code:`GraphWrapper`)
 
-        graph_level (default: False):  If :code:`graph_level=True` return shape (num_graphs, 1)
-                    elif :code:`graph_level=False return shape (num_nodes, 1)`
+        feature: A tensor with shape (num_nodes, hidden_size)
 
     Return:
-        A tensor with shape (num_graphs, 1) or (num_node, 1)
+        A tensor with shape (num_nodes, hidden_size)
     """
     nodes = fluid.layers.fill_constant(
         [gw.num_nodes, 1], dtype="float32", value=1.0)
-- 
GitLab