torch.lgamma

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

Вычисляет натуральный логарифм абсолютного значения гамма-функции для input.

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