8344037789a11e0ec254b1d7aa8815131f5ecabd,torch_geometric/nn/dense/dense_gcn_conv.py,DenseGCNConv,forward,#DenseGCNConv#Any#Any#Any#Any#,32

Before Change

        out = torch.matmul(x, self.weight)
        deg = adj.sum(dim=-1, keepdim=True).clamp(min=1).pow(-0.5)
        deg_matrix = torch.bmm(deg, deg.permute(0, 2, 1))
        adj_deg = deg_matrix * adj
        out = torch.bmm(adj_deg, out) if B!=1 else torch.matmul(adj_deg, out)

        if self.bias is not None:

After Change

        out = torch.matmul(x, self.weight)
        deg_inv_sqrt = adj.sum(dim=-1).clamp(min=1).pow(-0.5)

        adj = deg_inv_sqrt.unsqueeze(-1) * adj * deg_inv_sqrt.unsqueeze(-2)
        out = torch.matmul(adj, out)

        if self.bias is not None:
            out = out + self.bias

Instances