jax.random.triangular#

jax.random.triangular(key, left, mode, right, shape=None, dtype=<class 'float'>)[source]#

採樣具有給定形狀和浮點 dtype 的三角分佈隨機值。

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

\[\begin{split}f(x; a, b, c) = \frac{2}{c-a} \left\{ \begin{array}{ll} \frac{x-a}{b-a} & a \leq x \leq b \\ \frac{c-x}{c-b} & b \leq x \leq c \end{array} \right.\end{split}\]

在域 \(a \leq x \leq c\) 上。

參數:

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

  • left (RealArray) – 一個浮點數或浮點數陣列，與 shape 廣播相容，表示分佈的下限參數。

  • mode (RealArray) – 一個浮點數或浮點數陣列，與 shape 廣播相容，表示分佈的峰值參數，值必須滿足條件 left <= mode <= right。

  • right (RealArray) – 一個浮點數或浮點數陣列，與 shape 廣播相容，表示分佈的上限參數，必須大於 left。

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

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

返回:

一個具有指定 dtype 的隨機陣列，其形狀由 shape 給出（如果 shape 不是 None），否則由 left.shape、mode.shape 和 right.shape 給出。

返回類型:

Array