polytopax.generate_direction_vectors

polytopax.generate_direction_vectors(dimension: int, n_directions: int, method: Literal['uniform', 'icosphere', 'adaptive'] = 'uniform', random_key: Array | None = None) → Array[source]

Generate direction vectors for sampling.

Parameters:

  • dimension – Spatial dimension

  • n_directions – Number of directions to generate

  • method – Sampling strategy - “uniform”: Uniform distribution on sphere - “icosphere”: Icosahedral subdivision (3D only) - “adaptive”: Locally adaptive density sampling

  • random_key – JAX random key (required for “uniform” and “adaptive”)

Returns:

Normalized direction vector set with shape (n_directions, dimension)

Raises:

ValueError – Invalid parameters or unsupported combinations