94c66f7bf013faf82184a1b8c3c61a17f5af3882,torch_geometric/transform/graclus.py,,cluster_adj,#Any#Any#,92

Before Change

    weight = adj._values()
    weight[row == col] = 0
    adj = torch.sparse.FloatTensor(index, weight, torch.Size([n, n]))
    return adj.coalesce()


def cluster_position(pos, cluster, singleton):
    dim = pos.size(1)

After Change

    row, col = adj._indices()
    row, col = cluster[row], cluster[col]
    weight = adj._values()
    mask = row != col
    row, col, weight = row[mask], col[mask], weight[mask]
    index = torch.stack([row, col], dim=0)
    adj = torch.sparse.FloatTensor(index, weight, torch.Size([n, n]))
    return adj

Instances