spaces.GridAxis

spaces.GridAxis(low, high, n, shape=None)

Axis for systematic grid sampling over parameter bounds.

Generates linearly spaced values between low and high bounds using deterministic grid sampling.

Parameters

Name Type Description Default
low float Lower bound for sampling. required
high float Upper bound for sampling. required
n int Number of grid points to generate. required

Raises

Name Type Description
ValueError If n <= 0 or low >= high.

Examples

>>> grid = GridAxis(0.0, 1.0, 5)
>>> values = grid.generate_values()
>>> print(values)  # [0.0, 0.25, 0.5, 0.75, 1.0]

Attributes

Name Description
size Number of grid points.

Methods

Name Description
generate_values Generate linearly spaced grid values.

generate_values

spaces.GridAxis.generate_values(key=None)

Generate linearly spaced grid values.

Parameters

Name Type Description Default
key jax.random.PRNGKey Random key. Ignored for deterministic grid sampling. None

Returns

Name Type Description
jnp.ndarray Array of linearly spaced values from low to high. If shape is specified, values are broadcast to shape (n,) + shape with identical values across additional dimensions.