pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
21.46k stars 3.68k forks source link

Fix `utils.group_cat` concatenating dimension #9766

Closed tzuhanchang closed 1 week ago

tzuhanchang commented 2 weeks ago

Before the fix, tensors can only be concatenated over dim=0. The dim argument was not used by any operation in the function. This update allows the tensors to be concatenated over any given dimension.