Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!
Get Started for Free

torch_polygamma

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 ≥q 0 is called the order of the polygamma function.

ψ^{(n)}(x) = \frac{d^{(n)}}{dx^{(n)}} ψ(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

Tensors and Neural Networks with 'GPU' Acceleration

v0.3.0
MIT + file LICENSE
Authors
Daniel Falbel [aut, cre, cph], Javier Luraschi [aut], Dmitriy Selivanov [ctb], Athos Damiani [ctb], Christophe Regouby [ctb], Krzysztof Joachimiak [ctb], RStudio [cph]
Initial release

We don't support your browser anymore

Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.