Open3D (C++ API)  0.18.0+252c867
Macros | Functions | Variables
FixedRadiusSearchOps.cpp File Reference

(252c867 (Sun May 5 05:39:34 2024 +0000))

#include <vector>
#include "open3d/core/Dtype.h"
#include "open3d/core/nns/NeighborSearchCommon.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/utility/Helper.h"
#include "torch/script.h"

Macros

#define FN_PARAMETERS
 

Functions

template<class T , class TIndex >
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, torch::ScalarType index_dtype, const std::string &metric_str, const bool ignore_query_point, const bool return_distances)
 

Variables

const char * fixed_radius_fn_format
 

Macro Definition Documentation

◆ FN_PARAMETERS

#define FN_PARAMETERS
Value:
points, queries, radius, points_row_splits, queries_row_splits, \
hash_table_splits, hash_table_index, hash_table_cell_splits, \
metric, ignore_query_point, return_distances, neighbors_index, \
neighbors_row_splits, neighbors_distance
int points
Definition: FilePCD.cpp:54

Function Documentation

◆ FixedRadiusSearch()

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,
torch::ScalarType  index_dtype,
const std::string &  metric_str,
const bool  ignore_query_point,
const bool  return_distances 
)

◆ FixedRadiusSearchCPU()

template<class T , class TIndex >
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 
)

Variable Documentation

◆ fixed_radius_fn_format

const char* fixed_radius_fn_format
Initial value:
=
"open3d::fixed_radius_search(Tensor points, Tensor queries, float "
"radius, Tensor points_row_splits, Tensor queries_row_splits, Tensor "
"hash_table_splits, Tensor hash_table_index, Tensor "
"hash_table_cell_splits, ScalarType index_dtype=%d, str "
"metric=\"L2\", "
"bool ignore_query_point="
"False, bool return_distances=False"
") -> (Tensor neighbors_index, "
"Tensor neighbors_row_splits, Tensor neighbors_distance)"