jax.random.gamma#

jax.random.gamma(key, a, shape=None, dtype=<class 'float'>)[原始碼]#

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

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

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

在 \(0 \le x < \infty\) 域上，且 \(a > 0\)。

這是標準 Gamma 密度，具有單位尺度/速率參數。將樣本輸出除以速率等同於從 gamma(a, rate) 取樣，而將樣本輸出乘以尺度等同於從 gamma(a, scale) 取樣。

參數:

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

  • a (RealArray) – 與 shape 廣播相容的浮點數或浮點數陣列，表示分佈的參數。

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

  • dtype (DTypeLikeFloat) – 選填，用於傳回值的浮點數 dtype (如果 jax_enable_x64 為 true，則預設為 float64，否則為 float32)。

傳回值:

具有指定 dtype 且形狀由 shape (如果 shape 不是 None) 或 a.shape 給定的隨機陣列。

傳回型別:

Array

另請參閱

loggamma在對數空間中取樣 gamma 值，這可以為小的 a 值提供更佳的

準確度。