about gnn module compute neighbor features

[yyyyy1231]

I see this "# do this for every sample in batch, not nice, but I don't know how to use index_select batchwise" in model code.Why gnn part need batchwise to compute neighbor features?Thx a lot.