torch (version 0.2.0)

torch_polygamma: Polygamma

Description

Polygamma

Usage

torch_polygamma(n, self)

Arguments

n

(int) the order of the polygamma function

self

(Tensor) the input tensor.

polygamma(n, input, out=NULL) -> Tensor

Computes the \(n^{th}\) derivative of the digamma function on input. \(n \geq 0\) is called the order of the polygamma function.

$$ \psi^{(n)}(x) = \frac{d^{(n)}}{dx^{(n)}} \psi(x) $$

Examples

# NOT RUN {
if (torch_is_installed()) {
# }
# NOT RUN {
a = torch_tensor(c(1, 0.5))
torch_polygamma(1, a)
# }
# NOT RUN {
}
# }