2f62e4fb3470dc71fde45415d7c80c1e037de7fb,torch_geometric/utils/coalesce.py,,coalesce,#Any#Any#,5

Before Change

def coalesce(edge_index, num_nodes=None):
    num_nodes = edge_index.max().item() + 1 if num_nodes is None else num_nodes
    row, col = edge_index

    index = num_nodes * row + col
    perm = torch.arange(index.size(0), dtype=torch.long, device=row.device)

After Change

def coalesce(edge_index, num_nodes=None):
    num_nodes = get_num_nodes(edge_index, num_nodes)
    row, col = edge_index

    index = num_nodes * row + col
    perm = torch.arange(index.size(0), dtype=torch.long, device=row.device)

Instances