Function torch::special::polygamma¶
Defined in File special.h
Function Documentation¶
-
inline Tensor torch::special::polygamma(int64_t n, const Tensor &self)¶
Computes the nth derivative of the digamma function on the input.
See https:://pytorch.org/docs/main/special.html#torch.special.polygamma.
Example:
auto t = torch::randn(128, dtype=kDouble); torch::special::polygamma(2, t);