open3d.core.nns.NearestNeighborSearch

class open3d.core.nns.NearestNeighborSearch

NearestNeighborSearch class for nearest neighbor search. Construct a NearestNeighborSearch object with input dataset_points of shape {n_dataset, d}.

__init__(self: open3d.cpu.pybind.core.nns.NearestNeighborSearch, dataset_points: open3d.cpu.pybind.core.Tensor, index_dtype: open3d.cpu.pybind.core.Dtype = Int64) → None

fixed_radius_index(self: open3d.cpu.pybind.core.nns.NearestNeighborSearch, radius: Optional[float] = None) → bool

fixed_radius_search(self, query_points, radius, sort=None)

Parameters

• query_points (open3d.core.Tensor) – The query tensor of shape {n_query, d}.

• radius (float) – Radius value for radius search.

• sort (Optional[bool], optional, default=None) –

Returns

Tuple[open3d.core.Tensor, open3d.core.Tensor, open3d.core.Tensor]

hybrid_index(self: open3d.cpu.pybind.core.nns.NearestNeighborSearch, radius: Optional[float] = None) → bool

hybrid_search(self, query_points, radius, max_knn)

Perform hybrid search.

Parameters

• query_points (open3d.core.Tensor) – The query tensor of shape {n_query, d}.

• radius (float) – Radius value for radius search.

• max_knn (int) – Maximum number of neighbors to search per query point.

Returns

Tuple[open3d.core.Tensor, open3d.core.Tensor, open3d.core.Tensor]

knn_index(self: open3d.cpu.pybind.core.nns.NearestNeighborSearch) → bool

Set index for knn search.

knn_search(self, query_points, knn)

Perform knn search.

Parameters

• query_points (open3d.core.Tensor) – The query tensor of shape {n_query, d}.

• knn (int) – Number of neighbors to search per query point.

Returns

Tuple[open3d.core.Tensor, open3d.core.Tensor]

multi_radius_index(self: open3d.cpu.pybind.core.nns.NearestNeighborSearch) → bool

Set index for multi-radius search.

multi_radius_search(self, query_points, radii)

Perform multi-radius search. Each query point has an independent radius.

Parameters

• query_points (open3d.core.Tensor) – The query tensor of shape {n_query, d}.

• radii (open3d.core.Tensor) – Tensor of shape {n_query,} containing multiple radii, one for each query point.

Returns

Tuple[open3d.core.Tensor, open3d.core.Tensor, open3d.core.Tensor]