jax.random.geometric

jax.random.geometric(key, p, shape=None, dtype=<class 'int'>)

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

這些值根據機率質量函數返回

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

在 \(0 < p < 1\) 域上。

參數:

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

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

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

• dtype (DTypeLikeInt) – 選填，返回值的整數 dtype (如果 jax_enable_x64 為 true，則預設為 int64，否則為 int32)。

返回:

具有指定 dtype 的隨機陣列，其形狀由 shape 給定 (如果 shape 不是 None)，否則由 p.shape 給定。

返回類型:

陣列