jax.lax.rng_uniform

jax.lax.rng_uniform(a, b, shape)

有状态 PRNG 生成器。实验性的，不鼓励使用。

返回范围 [a, b) 内均匀分布的随机数。如果 b <= a，则结果未定义，并且不同的实现可能会返回不同的结果。

对于大多数用途，您应该使用 jax.random；此函数仅适用于具有特殊性能要求的利基用例。

此 API 可能会随时删除。