jax.random.dirichlet

jax.random.dirichlet(key, alpha, shape=None, dtype=<class 'float'>)

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

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

\[f(\{x_i\}; \{\alpha_i\}) \propto \prod_{i=1}^k x_i^{\alpha_i - 1}\]

其中 \(k\) 是維度，且 \(\{x_i\}\) 滿足

\[\sum_{i=1}^k x_i = 1\]

且 \(0 \le x_i \le 1\) 對於所有 \(x_i\)。

參數:

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

• alpha (RealArray) – 形狀為 (..., n) 的陣列，用作隨機變數的濃度參數。

• shape (Shape | None | None) – 選用，指定結果批次形狀的非負整數元組；也就是說，結果形狀的前綴，不包括值為 n 的最後一個元素。必須與 alpha.shape[:-1] 廣播相容。預設值 (None) 會產生與 alpha.shape 相等的结果形狀。

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

傳回:

一個隨機陣列，其指定的 dtype 和形狀由 shape + (alpha.shape[-1],) (如果 shape 不是 None) 或 alpha.shape 給定。

傳回型別:

Array