neuralop.layers.neighbor_search.native_neighbor_search

neuralop.layers.neighbor_search.native_neighbor_search(data: Tensor, queries: Tensor, radius: float, return_norm: bool = False)[source]

Native PyTorch implementation of a neighborhood search between two arbitrary coordinate meshes.

Parameters:

datatorch.Tensor

vector of data points from which to find neighbors

queriestorch.Tensor

centers of neighborhoods

radiusfloat

size of each neighborhood