jax.random.loggamma

jax.random.loggamma(key, a, shape=None, dtype=<class 'float'>)

採樣具有給定形狀和浮點 dtype 的 log-gamma 隨機值。

此函數的實作方式使得以下條件在 dtype 適當的容差下成立

np.testing.assert_allclose(jnp.exp(loggamma(*args)), gamma(*args), rtol=rtol)

log-gamma 的優點在於，對於非常接近零的樣本（當 a << 1 時經常發生），在對數空間中採樣可提供更高的精度。

參數:

• 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 不是 None，則形狀由 shape 給出，否則由 a.shape 給出。

返回類型:

陣列

另請參閱

gamma : 標準 gamma 採樣器。