Open3D (C++ API)
0.12.0
|
#include <vector>
#include "open3d/ml/impl/misc/NeighborSearchCommon.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "torch/script.h"
Macros | |
#define | FN_PARAMETERS |
#define | CALL(type, fn) |
Functions | |
template<class T > | |
void | FixedRadiusSearchCPU (const torch::Tensor &points, const torch::Tensor &queries, double radius, const torch::Tensor &points_row_splits, const torch::Tensor &queries_row_splits, const torch::Tensor &hash_table_splits, const torch::Tensor &hash_table_index, const torch::Tensor &hash_table_cell_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) |
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor > | FixedRadiusSearch (torch::Tensor points, torch::Tensor queries, double radius, torch::Tensor points_row_splits, torch::Tensor queries_row_splits, torch::Tensor hash_table_splits, torch::Tensor hash_table_index, torch::Tensor hash_table_cell_splits, const std::string &metric_str, const bool ignore_query_point, const bool return_distances) |
#define CALL | ( | type, | |
fn | |||
) |
#define FN_PARAMETERS |
std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> FixedRadiusSearch | ( | torch::Tensor | points, |
torch::Tensor | queries, | ||
double | radius, | ||
torch::Tensor | points_row_splits, | ||
torch::Tensor | queries_row_splits, | ||
torch::Tensor | hash_table_splits, | ||
torch::Tensor | hash_table_index, | ||
torch::Tensor | hash_table_cell_splits, | ||
const std::string & | metric_str, | ||
const bool | ignore_query_point, | ||
const bool | return_distances | ||
) |
void FixedRadiusSearchCPU | ( | const torch::Tensor & | points, |
const torch::Tensor & | queries, | ||
double | radius, | ||
const torch::Tensor & | points_row_splits, | ||
const torch::Tensor & | queries_row_splits, | ||
const torch::Tensor & | hash_table_splits, | ||
const torch::Tensor & | hash_table_index, | ||
const torch::Tensor & | hash_table_cell_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 | ||
) |