Install Theory Guide User Guide API Examples Developer's Guide
Github
  • Installing NeuralOperator
  • Theory Guide
  • User Guide
  • API reference
    • Models
    • Layers
    • Model Dispatching
    • Training
    • Data
  • Examples
  • Development guide

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

neuralop.layers.neighbor_search.NeighborSearch neuralop.layers.padding.DomainPadding
© Copyright 2025, Jean Kossaifi, David Pitt, Nikola Kovachki, Zongyi Li and Anima Anandkumar.

On this page

  • neuralop.layers.neighbor_search.native_neighbor_search
    • native_neighbor_search()