PyTorch Extension Library of Optimized Graph Cluster Algorithms
MIT License
Bot releases are visible (Hide)
torch.jit.script
conversion of functions on module importPublished by rusty1s about 1 year ago
batch_size
argument for fps
, knn
, radius
functions (https://github.com/rusty1s/pytorch_cluster/pull/175)ptr
argument https://github.com/rusty1s/pytorch_cluster/pull/180)Full Changelog: https://github.com/rusty1s/pytorch_cluster/compare/1.6.1...1.6.2
Published by rusty1s over 1 year ago
random_walk
(#139)bf16
support for knn
, radius
and graclus
(#144)batch
layout in nearest
(#168)Published by rusty1s over 2 years ago
num_nodes
in random_walk
(in case num_nodes=None
) (#112)knn
/radius
calculation for batches with zero-point examplesknn
/radius
calculationPublished by rusty1s over 3 years ago
fps
can now take in different ratios across different batched point sets, i.e. ratio
can now be a torch.Tensor
radius
computed slightly different results across CPU and CUDA versionsPublished by rusty1s almost 4 years ago
random_walk
now supports q != 1
and p != 1
via rejection samplingPublished by rusty1s about 4 years ago
radius
where the max_num_neighbors
argument has been ignoredPublished by rusty1s over 4 years ago
This release fixes some issues in the new knn
and radius
CPU implementations that have led to memory leaks. It is strongly recommended to update the package in case you are currently using torch-cluster==1.5.5
.
Published by rusty1s over 4 years ago
torch-cluster
is now fully-jittable thanks to new implementations for knn
and radius
based on nanoflann
rather than scipy
.Published by rusty1s over 4 years ago
Fixed a bug in the CUDA version of fps
.
Published by rusty1s over 4 years ago
Fixes regarding proper usage of fps
and radius
.
Published by rusty1s over 4 years ago
This release introduces traceable torch-cluster
operators using torch.jit.script
. As a result, PyTorch 1.4 is now required to install this package. We also provide Python wheels for common OS/CUDA combinations (see README).
Published by rusty1s about 5 years ago
This release brings PyTorch 1.3 support to torch-cluster
.
Published by rusty1s about 5 years ago
Published by rusty1s about 5 years ago
Published by rusty1s over 5 years ago
Added flow
argument to the radius_graph
method.
Published by rusty1s over 5 years ago
flow
argument to the knn_graph
method.Published by rusty1s over 5 years ago
Published by rusty1s over 5 years ago
Published by rusty1s over 5 years ago