09e572cef1dc655bfc5255ecd0b3787512609e3d,torch_geometric/nn/functional/graph_conv.py,,graph_conv,#Any#Any#Any#Any#Any#,27

Before Change

    row, col = edge_index
    n, e = x.size(0), row.size(0)

    if edge_attr is None:
        edge_attr = x.data.new(e).fill_(1)

    // Compute degree.
    degree = x.data.new(n).fill_(0).scatter_add_(0, row, edge_attr) + 1
    degree = degree.pow_(-0.5)

    // Normalize and append adjacency matrix by self loops.

After Change

    row, col = edge_index
    num_nodes, e = x.size(0), row.size(0)

    edge_attr = x.new_full((e, ), 1) if edge_attr is None else edge_attr
    deg = degree(row, num_nodes, dtype=x.dtype, device=x.device).pow_(-0.5)

    // Normalize and append adjacency matrix by self loops.
    edge_attr = deg[row] * edge_attr * deg[col]

Instances