jax.scipy.stats.multivariate_normal.pdf

jax.scipy.stats.multivariate_normal.pdf(x, mean, cov)

多變量常態機率分佈函數。

JAX 實作的 scipy.stats.multivariate_normal pdf。

多變量常態 PDF 定義如下：

\[f(x) = \frac{1}{(2\pi)^k\det\Sigma}\exp\left(-\frac{(x-\mu)^T\Sigma^{-1}(x-\mu)}{2} \right)\]

其中 \(\mu\) 是 mean，\(\Sigma\) 是共變異數矩陣 (cov)，而 \(k\) 是 \(\Sigma\) 的秩。

參數:

• x (Array | ndarray | bool | number | bool | int | float | complex) – 類陣列，用於評估 PDF 的值

• mean (Array | ndarray | bool | number | bool | int | float | complex) – 類陣列，分佈的質心

• cov (Array | ndarray | bool | number | bool | int | float | complex) – 類陣列，分佈的共變異數矩陣

• allow_singular – 不支援

傳回:

pdf 值的陣列。

傳回型別:

Array

另請參閱

jax.scipy.stats.multivariate_normal.logpdf()