top_k_graph() function not handling batched data?

HongyangGao / Graph-U-Nets

Pytorch implementation of Graph U-Nets (ICML19)

GNU General Public License v3.0

513 stars 100 forks source link

Open DanielGetter opened 6 months ago

DanielGetter commented 6 months ago

How can one ensure that indexing the h node matrix with the topk scores applies to each member of a batch of graph data?