dadin4vcc

### 🐛 Describe the bug Hi, Thanks for the great work with PyG! As reported in [this issue](https://github.com/pyg-team/pytorch_geometric/issues/4906), the knn_graph function gives too few neighbours (possiby only on GPU). It...