jax.random.binomial

jax.random.binomial(key, n, p, shape=None, dtype=<class 'float'>)

取樣具有給定形狀和浮點數 dtype 的二項分配隨機值。

這些值根據以下機率質量函數回傳

\[f(k;n,p) = \binom{n}{k}p^k(1-p)^{n-k}\]

在 \(0 < p < 1\) 的定義域上，其中 \(n\) 是表示試驗次數的非負整數，而 \(p\) 是表示單次試驗成功的機率的浮點數。

參數：

• key (Array) – 作為隨機金鑰使用的 PRNG 金鑰。

• n (RealArray) – 與 shape 廣播相容的浮點數或浮點數陣列，表示試驗次數。

• p (RealArray) – 與 shape 廣播相容的浮點數或浮點數陣列，表示單次試驗成功的機率。

• shape (Shape | None | None) – 選項性，指定結果形狀的非負整數元組。必須與 n 和 p 廣播相容。預設值 (None) 產生的結果形狀等於 np.broadcast(n, p).shape。

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

回傳：

具有指定 dtype 和形狀的隨機陣列，形狀由 np.broadcast(n, p).shape 給定。

回傳型別：

Array