torch_polygamma {torch} | R Documentation |
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 derivative of the digamma function on
input
.
is called the order of the polygamma function.
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)
}
[Package torch version 0.13.0 Index]