jax.random.rayleigh#
- jax.random.rayleigh(key, scale, shape=None, dtype=<class 'float'>)[來源]#
取樣具有指定形狀和浮點數資料類型的 Rayleigh 隨機值。
這些值根據以下機率密度函數回傳
\[f(x;\sigma) \propto xe^{-x^2/(2\sigma^2)}\]在定義域 \(-\infty < x < \infty\) 上,其中 \(\sigma > 0\) 是分佈的尺度參數。
- 參數:
key (ArrayLike) – 用作隨機金鑰的 PRNG 金鑰。
scale (RealArray) – 與
shape
廣播相容的浮點數或浮點數陣列,表示分佈的參數。shape (Shape | None | None) – 選填,指定結果形狀的非負整數元組。必須與
scale
廣播相容。預設值 (None) 會產生與scale.shape
相等的結果形狀。dtype (DTypeLikeFloat) – 選填,回傳值的浮點數資料類型 (如果 jax_enable_x64 為 true,則預設為 float64,否則為 float32)。
- 回傳:
具有指定資料類型和形狀的隨機陣列,如果
shape
不是 None,則形狀由shape
給定,否則由scale.shape
給定。- 回傳類型: