From 7ba8f3c78cc29619edcc7c710465fb5644ad6dec Mon Sep 17 00:00:00 2001 From: Illia Ovcharenko Date: Tue, 12 Nov 2024 19:08:00 +0200 Subject: [PATCH] change ratio type to tensor --- csrc/cluster.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/csrc/cluster.h b/csrc/cluster.h index ac8315b..e5a3b85 100644 --- a/csrc/cluster.h +++ b/csrc/cluster.h @@ -10,7 +10,7 @@ CLUSTER_INLINE_VARIABLE int64_t _cuda_version = cuda_version(); } // namespace detail } // namespace cluster -CLUSTER_API torch::Tensor fps(torch::Tensor src, torch::Tensor ptr, double ratio, +CLUSTER_API torch::Tensor fps(torch::Tensor src, torch::Tensor ptr, torch::Tensor ratio, bool random_start); CLUSTER_API torch::Tensor graclus(torch::Tensor rowptr, torch::Tensor col,