jax.lax.rng_uniform

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

具狀態的 PRNG 產生器。實驗性質，不建議使用。

傳回範圍 [a, b) 內均勻分佈的隨機數。如果 b <= a，則結果未定義，且不同的實作可能會傳回不同的結果。

您應該將 jax.random 用於大多數目的；此函式僅適用於具有特殊效能要求的利基用例。

此 API 可能隨時移除。