Skip to content

Commit

Permalink
Update layers.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Fengchuan-Song authored May 14, 2023
1 parent 938ed36 commit b09cd8e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pygcn/layers.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ def reset_parameters(self):

def forward(self, input, adj):
support = torch.mm(input, self.weight)
# Sparse before, Dense later
# Sparse before, Dense later, multiplication
output = torch.spmm(adj, support)
if self.bias is not None:
return output + self.bias
Expand Down

0 comments on commit b09cd8e

Please sign in to comment.