torch_polygamma: Polygamma

In torch: Tensors and Neural Networks with 'GPU' Acceleration

Polygamma

Description

Polygamma

Usage

torch_polygamma(n, input)

Arguments

n	(int) the order of the polygamma function
input	(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)

Note

This function is not implemented for \eqn{n \geq 2}.

Examples

if (torch_is_installed()) {
## Not run: 
a = torch_tensor(c(1, 0.5))
torch_polygamma(1, a)

## End(Not run)
}

torch documentation built on June 7, 2023, 6:19 p.m.