d83b62938607378a935bf657d782ded941f1577c,torch_geometric/transforms/gdc.py,GDC,_transition_matrix,#GDC#Any#Any#Any#Any#,95

Before Change

        diag_idx = diag_idx.unsqueeze(0).repeat(2, 1)

        if normalization == "sym":
            D_vec = torch.sparse.sum(sp_adj_matrix, dim=0).to_dense()
            D_vec_invsqrt = 1 / torch.sqrt(D_vec)
            edge_index, edge_weight = spspmm(diag_idx, D_vec_invsqrt,
                                             edge_index, edge_weight,
                                             num_nodes, num_nodes, num_nodes)

After Change

                                             num_nodes, num_nodes, num_nodes)
        elif normalization == "col":
            _, col = edge_index
            D_vec = scatter_add(edge_weight, col, dim=0, dim_size=num_nodes)
            D_vec_inv = 1 / D_vec
            edge_index, edge_weight = spspmm(edge_index, edge_weight,
                                             diag_idx, D_vec_inv,
                                             num_nodes, num_nodes, num_nodes)

Instances