neuralop.layers.neighbor_search.NeighborSearch

class neuralop.layers.neighbor_search.NeighborSearch(use_open3d=True)[source]

Neighborhood search between two arbitrary coordinate meshes. For each point x in queries, returns a set of the indices of all points y in data within the ball of radius r B_r(x)

Parameters:

use_open3dbool

Whether to use open3d or native PyTorch implementation NOTE: open3d implementation requires 3d data

Methods

forward(data, queries, radius)

Find the neighbors, in data, of each point in queries within a ball of radius.

forward(data, queries, radius)[source]

Find the neighbors, in data, of each point in queries within a ball of radius. Returns in CRS format.

Parameters:

datatorch.Tensor of shape [n, d]

Search space of possible neighbors NOTE: open3d requires d=3

queriestorch.Tensor of shape [m, d]

Points for which to find neighbors NOTE: open3d requires d=3

radiusfloat

Radius of each ball: B(queries[j], radius)