Skip to content

GitLab

Unverified Commit 6af55f18 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

Update gcn2conv.py (#6020) 


Co-authored-by: default avatarHongzhi (Steve), Chen <chenhongzhi.nkcs@gmail.com>
Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent e37c8acb
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 22 deletions
python/dgl/nn/pytorch/conv/gcn2conv.py
View file @ 6af55f18
......@@ -256,29 +256,12 @@ class GCN2Conv(nn.Module):
feat_0 = feat_0[: feat.size(0)] * self.alpha
if self._project_initial_features:
rst = feat.add_(feat_0)
rst = th.addmm(
feat,
feat,
self.weight1,
beta=(1 - self.beta),
alpha=self.beta,
)
feat_sum = feat + feat_0
rst = (1 - self.beta) * feat_sum + self.beta * feat_sum @ self.weight1
else:
rst = th.addmm(
feat,
feat,
self.weight1,
beta=(1 - self.beta),
alpha=self.beta,
)
rst += th.addmm(
feat_0,
feat_0,
self.weight2,
beta=(1 - self.beta),
alpha=self.beta,
)
feat_sum = feat + feat_0
feat_proj_sum = feat @ self.weight1 + feat_0 @ self.weight2
rst = (1 - self.beta) * feat_sum + self.beta * feat_proj_sum
if self._bias:
rst = rst + self.bias
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment