boax.core.samplers.uniform

Contents

boax.core.samplers.uniform#

boax.core.samplers.uniform(key, sample_shape)#

The i.i.d. uniform sampler.

Example

>>> sampler = uniform(key, (128,))
>>> samples = sampler(Uniform(0, 1))
Parameters:

  • key (Array) – The pseudo-random number genererator key.

  • sample_shape (Sequence[Union[int, Any]]) – The sample shape.

Return type:

Sampler[Uniform]

Returns:

The corresponding Sampler.

Contents