fastdev.geom.sampling¶

Module Contents¶

fastdev.geom.sampling.sample_farthest_points(points: jaxtyping.Float[torch.Tensor, *B N 3], num_samples: int, random_start: bool = False) → jaxtyping.Int64[torch.Tensor, *B num_samples][source]¶

Sample farthest points.

Parameters:

points (Tensor) – input points in shape (B, N, 3) or (N, 3)
num_samples (int) – number of samples
random_start (bool)

Returns:

indices of farthest points in shape (B, num_samples) or (num_samples,)

Return type:

Tensor