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]])