jax.random.wald

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

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

這些值根據以下機率密度函數回傳

\[f(x;\mu) = \frac{1}{\sqrt{2\pi x^3}} \exp\left(-\frac{(x - \mu)^2}{2\mu^2 x}\right)\]

在域 \(-\infty < x < \infty\) 上，其中 \(\mu > 0\) 是分布的位置參數。

參數:

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

• mean (RealArray) – 與 shape 廣播相容的浮點數或浮點數陣列，表示分布的平均值參數。

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

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

回傳值:

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

回傳類型:

Array