jax.random.exponential

jax.random.exponential(key, shape=(), dtype=<class 'float'>)

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

這些值根據以下機率密度函數分佈

\[f(x) = e^{-x}\]

在定義域 \(0 \le x < \infty\) 上。

參數:

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

• shape (Shape) – 選填，表示結果形狀的非負整數 tuple。預設值為 ()。

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

返回:

具有指定形狀和 dtype 的隨機陣列。

返回型別:

Array