d8a075668b6e9cdf4c08f6c7285e5c7d2fbf5332,torch_geometric/graph/geometry.py,,edges_from_faces,#Any#,38

Before Change

    // Sort the adjacencies row-wise.
    edges = edges.t()
    sorted, indices = torch.sort(edges[0], 0)
    edges = torch.cat((sorted, edges[1][indices])).view(2, -1)
    return edges

After Change

def edges_from_faces(faces):
    // Get directed edges.
    edges = torch.cat((faces[:, 0:2], faces[:, 1:3], faces[:, 0:3:2]))
    n = faces.max() + 1

    // Build directed adjacency matrix.
    adj = torch.sparse.FloatTensor(edges.t(),
                                   torch.ones(edges.size(0)),
                                   torch.Size([n, n]))

    // Convert to undirected adjacency matrix.
    adj = adj + adj.t()

    // Remove duplicate indices.
    // NOTE: This doesn"t work if transpose(...) is removed.
    adj = adj.transpose(0, 1).coalesce()
    return adj._indices()

Instances