pytorch_cluster
pytorch_cluster copied to clipboard
PyTorch Extension Library of Optimized Graph Cluster Algorithms
Variable K in KNN
Thank you very much for developing such a great library. Currently, I have a problem that is bothering me. In KNN, if I want to choose the number of neighbors...
[MPS] PyTorch Nightly with MPS on Mac M1 does not work
Hi, I am currently trying to get pytorch geometric to run on my Mac with M1 Pro, the recent update to 13.3 fixed quite a few things. Look here for...
General Union types are not currently supported in v1.6.1
RuntimeError: General Union types are not currently supported. Only Union[T, NoneType] (i.e. Optional[T]) is supported.: File "/home/zj/miniconda3/envs/yukewang/lib/python3.8/site-packages/torch_cluster/rw.py", line 18 num_nodes: Optional[int] = None, return_edge_indices: bool = False, ) -> Union[Tensor,...
Infinite loop in torch_cluster.graclus_cluster on empty inputs?
When calling `graclus_cluster(row, col, weight, numnodes)` on empty input, the call hangs. In particular by "empty inputs" I mean inputs like: `row=torch.Tensor([])`, `col=torch.Tensor([])`, `weight=torch.Tensor([])`, `numnodes=some positive integer`. While the call...
`radius_graph` bug: it might produce `max_num_neighbors + 1` neighbors for both CPU and GPU versions
Hi Team, I found that `radius_graph` might produce `max_num_neighbors + 1` neighbors for both CPU and GPU versions. A minimal reproducer for CPU, adapted from the example in README: ```python...
The current implementation of farthest point sampling only allows to set a ratio value to define the number of output points. Some use-cases require a fixed number of output points....
When executing torch_cluster.radius on my MacBook Pro 16-inch(Nov 2023, M3 Pro) even a minimal working example gives an error message. Problem does NOT persist on Server Ubuntu 22.04.4. Minimum working...
When install torch-cluster using the command ```bash python3 -m pip install torch-cluster ``` I get the following error: ```bash Defaulting to user installation because normal site-packages is not writeable Collecting...
In conda ven, `python setup.py install` but have sub err apper!! > /lib/python3.12/site-packages/torch/include/torch/csrc/jit/api/module.h:409:113: error: use of deleted function ‘torch::jit::Module::Module(torch::jit::Module&&)’ 409 | bool return_module) // if true include root itself as...
The Haversine (great-circle) distance metric would be IMO a very useful addition. It'd allow for better clustering of points residing on a sphere (particularly for point / node configurations where...