jax.random.beta#

jax.random.beta(key, a, b, shape=None, dtype=<class 'float'>)[source]#

採樣具有給定形狀和浮點數 dtype 的 Beta 隨機值。

這些值根據機率密度函數分佈

\[f(x;a,b) \propto x^{a - 1}(1 - x)^{b - 1}\]

在 \(0 \le x \le 1\) 域中。

參數：

  • key (ArrayLike) – 用作隨機金鑰的 PRNG 金鑰。

  • a (RealArray) – 一個浮點數或浮點數陣列，與 shape 廣播相容，代表第一個參數「alpha」。

  • b (RealArray) – 一個浮點數或浮點數陣列，與 shape 廣播相容，代表第二個參數「beta」。

  • shape (Shape | None | None) – 選項性，指定結果形狀的非負整數元組。必須與 a 和 b 廣播相容。預設值 (None) 會透過廣播 a 和 b 產生結果形狀。

  • dtype (DTypeLikeFloat) – 選項性，傳回值的浮點數 dtype（如果 jax_enable_x64 為 true，則預設為 float64，否則為 float32）。

傳回：

一個隨機陣列，具有指定的 dtype 和形狀，如果 shape 不是 None，則由 shape 給定，否則透過廣播 a 和 b 給定。

傳回型別：

陣列