torch (version 0.11.0)

torch_digamma: Digamma

Description

Digamma

Usage

torch_digamma(self)

Arguments

self

(Tensor) the tensor to compute the digamma function on

digamma(input, out=NULL) -> Tensor

Computes the logarithmic derivative of the gamma function on input.

$$\psi (x)=\frac{d}{dx}\ln \left(\mathrm{\Gamma }\left(x\right)\right)=\frac{{\mathrm{\Gamma }}^{\prime }(x)}{\mathrm{\Gamma }(x)}$$

Examples

if (torch_is_installed()) {

a = torch_tensor(c(1, 0.5))
torch_digamma(a)
}