emmi.utils.geometric

Functions

radius(x, y, r, max_num_neighbors, batch_x, batch_y)
knn(x, y, k[, batch_x, batch_y])
segment_csr(src, indptr, reduce)

Module Contents

emmi.utils.geometric.radius(x, y, r, max_num_neighbors, batch_x, batch_y)

emmi.utils.geometric.knn(x, y, k, batch_x=None, batch_y=None)

emmi.utils.geometric.segment_csr(src, indptr, reduce)