top_k_graph() function not handling batched data? #28
Description
How can one ensure that indexing the h node matrix with the topk scores applies to each member of a batch of graph data?
{{ message }}
How can one ensure that indexing the h node matrix with the topk scores applies to each member of a batch of graph data?