6f3f79792891b35abf0f9281a226dcd2e0a57eb3,torch_geometric/nn/dense/dense_graph_conv.py,DenseGraphConv,forward,#DenseGraphConv#Any#Any#Any#,25

Before Change

        elif self.aggr == "max":
            out = out.max(dim=-1, keepdim=True)[0]

        out = out + self.lin_root(x)

        if mask is not None:
            out = out * mask.view(B, N, 1).to(x.dtype)

After Change

            out[out == float("-inf")] = 0.

        out = self.lin_rel(out)
        out += self.lin_root(x)

        if mask is not None:
            out = out * mask.view(-1, N, 1).to(x.dtype)

Instances