(self, radius, num_neighbors, include_coordinates=True)
| 8 | |
| 9 | class BallQuery(nn.Module): |
| 10 | def __init__(self, radius, num_neighbors, include_coordinates=True): |
| 11 | super().__init__() |
| 12 | self.radius = radius |
| 13 | self.num_neighbors = num_neighbors |
| 14 | self.include_coordinates = include_coordinates |
| 15 | |
| 16 | def forward(self, points_coords, centers_coords, temb, points_features=None): |
| 17 | points_coords = points_coords.contiguous() |
nothing calls this directly
no outgoing calls
no test coverage detected