jax.lax.rng_uniform# jax.lax.rng_uniform(a, b, shape)[source]# 有状态 PRNG 生成器。实验性，不建议使用。 返回范围 [a, b) 内的均匀分布随机数。如果 b <= a，则结果未定义，不同的实现可能会返回不同的结果。 您应该在大多数情况下使用 jax.random；此函数仅用于具有特殊性能要求的利基用例。 此 API 可能会随时删除。
