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

    singleton = cluster == -1
    num_singletons = singleton.sum()
    if num_singletons > 0:
        index = torch.arange(count, count + num_singletons).long()
        cluster[singleton] = index
        cluster_full = torch.cat([cluster, index], dim=0)
        return cluster, cluster_full, singleton
    else:
        return cluster, cluster, singleton

Instances