From 7eb8941ce2892b124abc8e18c3fe3edb7efbafd1 Mon Sep 17 00:00:00 2001
From: Webbley <liwb5@foxmail.com>
Date: Fri, 17 Apr 2020 15:36:43 +0800
Subject: [PATCH] fix bug of elementwise_mul error

---
 pgl/layers/conv.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pgl/layers/conv.py b/pgl/layers/conv.py
index 96e1bb9..bbb364a 100644
--- a/pgl/layers/conv.py
+++ b/pgl/layers/conv.py
@@ -230,7 +230,7 @@ def gin(gw,
         epsilon.stop_gradient = True
 
     msg = gw.send(send_src_copy, nfeat_list=[("h", feature)])
-    output = gw.recv(msg, "sum") + (1.0 + epsilon) * feature
+    output = gw.recv(msg, "sum") + feature * (epsilon + 1.0)
 
     output = fluid.layers.fc(output,
                              size=hidden_size,
-- 
GitLab