torch.lgamma

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

計算 input 中伽瑪函數絕對值的自然對數。

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

參數

input (Tensor) – 輸入張量。

關鍵字參數

out (Tensor, optional) – 輸出張量 (可選)。

範例

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