55 const Dtype &index_dtype) = 0;
64 const Dtype &index_dtype) = 0;
88 const Tensor &query_points,
103 const Tensor &query_points,
double radius,
bool sort)
const = 0;
116 const Tensor &query_points,
double radius,
int max_knn)
const = 0;
Definition: NNSIndex.h:40
Dtype index_dtype_
Definition: NNSIndex.h:140
Dtype GetDtype() const
Definition: NNSIndex.cpp:43
virtual std::tuple< Tensor, Tensor, Tensor > SearchHybrid(const Tensor &query_points, double radius, int max_knn) const =0
virtual ~NNSIndex()
Definition: NNSIndex.h:44
NNSIndex(const NNSIndex &)=delete
NNSIndex & operator=(const NNSIndex &)=delete
size_t GetDatasetSize() const
Definition: NNSIndex.cpp:38
Device GetDevice() const
Definition: NNSIndex.cpp:45
virtual std::tuple< Tensor, Tensor, Tensor > SearchRadius(const Tensor &query_points, double radius, bool sort) const =0
virtual std::tuple< Tensor, Tensor, Tensor > SearchRadius(const Tensor &query_points, const Tensor &radii, bool sort) const =0
virtual bool SetTensorData(const Tensor &dataset_points, const Dtype &index_dtype)=0
virtual std::pair< Tensor, Tensor > SearchKnn(const Tensor &query_points, int knn) const =0
Tensor dataset_points_
Definition: NNSIndex.h:139
NNSIndex()
Default Constructor.
Definition: NNSIndex.h:43
Dtype GetIndexDtype() const
Definition: NNSIndex.cpp:47
int GetDimension() const
Definition: NNSIndex.cpp:33
virtual bool SetTensorData(const Tensor &dataset_points, double radius, const Dtype &index_dtype)=0
Definition: PinholeCameraIntrinsic.cpp:35