jax.lax.rng_uniform

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

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

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

您应该使用 jax.random 用于大多数目的；此函数仅存在于具有特殊性能要求的利基用例中。

此 API 可能会随时删除。