0ab5b005164cddccbf7c8a3d66ff6707333de80b,torch_geometric/utils/matmul.py,,matmul,#Any#Any#Any#,6

Before Change

    row = row if torch.is_tensor(tensor) else Variable(row)

    output_col = edge_attr.unsqueeze(-1) * tensor[col]
    output = new(output_col, num_nodes, dim).fill_(0)
    row_expand = row.unsqueeze(-1).expand_as(output_col)
    output.scatter_add_(0, row_expand, output_col)

    return output

After Change

    row, col = index
    out_col = value.unsqueeze(-1) * tensor[col]
    out = scatter_add(out_col, row, dim=0, dim_size=tensor.size(0))

    return out

Instances