differt.utils.sample_points_in_bounding_box

sample_points_in_bounding_box(bounding_box, shape=(), *, key)

Sample point(s) in a 3D bounding box.

Parameters:

• bounding_box (Float[ArrayLike, '2 3']) – The bounding box (min. and max. coordinates).

• shape (tuple[int, ...]) – The sample shape.

• key (Key[Array, ''] | UInt32[Array, '2']) – The jax.random.PRNGKey to be used.

Return type:

Float[Array, '*shape 3']

Returns:

An array of points randomly sampled.