torch.lgamma

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

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

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

Параметры

input (Тензор) – входной тензор.

Ключевые аргументы

out (Тензор, необязательно) – выходной тензор.

Пример:

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