Doubt related to the coalesce method

[yorard123]

Hi,

I would like to know if is it possible to make this method return the 'perm' tensor that contains the new order of the 'edge_index tensor'. I am asking that, because I see that the method allows to put as input the edge_attr and the edge_indx tensors, however, I have also a tensor related to the edges called edge_label, and I would like to filter that too (it can contain duplicated entries).

Thanks,

[wsad1]

You can pass both edge_label and edge_attr as a list and torch_geometric.utils.coalesce will filter both items in the list. It would look something like this.

edge_index, (edge_attr, edge_label) = coalesce(edge_index, edge_attr = [edge_attr,edge_label])