Python PyTorch multigammaln用法及代碼示例

本文簡要介紹python語言中 torch.special.multigammaln 的用法。

用法:

torch.special.multigammaln(input, p, *, out=None) → Tensor

參數：

• input(Tensor) -計算多元log-gamma函數的張量

• p(int) -維數

關鍵字參數：

out(Tensor,可選的) -輸出張量。

計算維度為 p 的 multivariate log-gamma function 元素，由下式給出

\log(\Gamma_{p}(a)) = C + \displaystyle \sum_{i=1}^{p} \log\left(\Gamma\left(a - \frac{i - 1}{2}\right)\right)

其中 C = \log(\pi) \times \frac{p (p - 1)}{4} 和 \Gamma(\cdot) 是 Gamma 函數。

所有元素都必須大於 \frac{p - 1}{2} ，否則會拋出錯誤。

例子：

>>> a = torch.empty(2, 3).uniform_(1, 2)
>>> a
tensor([[1.6835, 1.8474, 1.1929],
        [1.0475, 1.7162, 1.4180]])
>>> torch.special.multigammaln(a, 2)
tensor([[0.3928, 0.4007, 0.7586],
        [1.0311, 0.3901, 0.5049]])