5980bd68fcd65ffe90fe560ff7db3f01e861426e,torch_geometric/transform/graclus.py,,normalized_cut,#Any#Any#,42

Before Change

    // Append singleton values to the end.
    singleton = cluster == -1
    index = torch.arange(count, singleton.sum() + count).long()
    cluster[singleton] = index
    return cluster, torch.cat([cluster, index], dim=0), singleton

After Change

    while row.dim() > 0:
        print(row[0], col[0])
        if row[0] == col[0]:
            print("das darf nicht sein")
        cluster[row[0]] = count
        cluster[col[0]] = count

        mask = (row != row[0]) & (row != col[0])

Instances