jax.lax.rng_uniform

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

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

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

在大多数情况下，您应该使用 jax.random；此函数仅用于具有特殊性能要求的特定用例。

此 API 可能会随时被删除。