FixedRadiusSearchOps.cpp File Reference

(b9e049c (Sun Dec 31 11:36:26 2023 -0800))

#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

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)"