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 給定。

回傳類型:

Array