torch.lgamma

torch.lgamma(input, *, out=None) → Tensor

计算输入张量的伽玛函数绝对值的自然对数。

$\text{out}_{i} = \ln |\Gamma(\text{input}_{i})|$

参数

input (Tensor) – 需要输入的张量。

关键字参数

out (Tensor, 可选) – 指定输出张量。

示例：

>>> a = torch.arange(0.5, 2, 0.5)
>>> torch.lgamma(a)
tensor([ 0.5724,  0.0000, -0.1208])