Open3D (C++ API)
0.12.0
|
#include "open3d/ml/impl/misc/KnnSearch.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/misc/NeighborSearchAllocator.h"
#include "torch/script.h"
Macros | |
#define | INSTANTIATE(T) |
Functions | |
template<class T > | |
void | KnnSearchCPU (const torch::Tensor &points, const torch::Tensor &queries, const int64_t k, const torch::Tensor &points_row_splits, const torch::Tensor &queries_row_splits, const Metric metric, const bool ignore_query_point, const bool return_distances, torch::Tensor &neighbors_index, torch::Tensor &neighbors_row_splits, torch::Tensor &neighbors_distance) |
#define INSTANTIATE | ( | T | ) |
void KnnSearchCPU | ( | const torch::Tensor & | points, |
const torch::Tensor & | queries, | ||
const int64_t | k, | ||
const torch::Tensor & | points_row_splits, | ||
const torch::Tensor & | queries_row_splits, | ||
const Metric | metric, | ||
const bool | ignore_query_point, | ||
const bool | return_distances, | ||
torch::Tensor & | neighbors_index, | ||
torch::Tensor & | neighbors_row_splits, | ||
torch::Tensor & | neighbors_distance | ||
) |