:param centers_coords: coordinates of centers, FloatTensor[B, 3, M] :param points_coords: coordinates of points, FloatTensor[B, 3, N] :param radius: float, radius of ball query :param num_neighbors: int, maximum number of neighbors :return: neighb
(centers_coords, points_coords, radius, num_neighbors)
| 6 | |
| 7 | |
| 8 | def ball_query(centers_coords, points_coords, radius, num_neighbors): |
| 9 | """ |
| 10 | :param centers_coords: coordinates of centers, FloatTensor[B, 3, M] |
| 11 | :param points_coords: coordinates of points, FloatTensor[B, 3, N] |
| 12 | :param radius: float, radius of ball query |
| 13 | :param num_neighbors: int, maximum number of neighbors |
| 14 | :return: |
| 15 | neighbor_indices: indices of neighbors, IntTensor[B, M, U] |
| 16 | """ |
| 17 | centers_coords = centers_coords.contiguous() |
| 18 | points_coords = points_coords.contiguous() |
| 19 | return _backend.ball_query(centers_coords, points_coords, radius, num_neighbors) |
no outgoing calls
no test coverage detected