jax.random.truncated_normal#

jax.random.truncated_normal(key, lower, upper, shape=None, dtype=<class 'float'>)[source]#

採樣具有給定形狀和 dtype 的截斷標準常態隨機值。

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

\[f(x) \propto e^{-x^2/2}\]

在定義域 \(\rm{lower} < x < \rm{upper}\) 上。

參數:

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

  • lower (RealArray) – 一個浮點數或浮點數陣列，表示截斷的下限。必須與 upper 廣播相容。

  • upper (RealArray) – 一個浮點數或浮點數陣列，表示截斷的上限。必須與 lower 廣播相容。

  • shape (Shape | None | None) – 選填，一個非負整數元組，指定結果形狀。必須與 lower 和 upper 廣播相容。預設值 (None) 會透過廣播 lower 和 upper 來產生結果形狀。

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

回傳:

一個隨機陣列，其 dtype 和形狀由 shape 指定（如果 shape 不是 None），否則由廣播 lower 和 upper 指定。回傳開區間 (lower, upper) 中的值。

回傳型別:

Array