Function torch::special::digamma

• Defined in File special.h

Function Documentation

inline Tensor torch::special::digamma(const Tensor &self)

Computes the logarithmic derivative of the gamma function on input See https://pytorch.org/docs/main/special.html#torch.special.digamma.

Example:

auto t = torch::randn(128, dtype=kDouble);
torch::special::digamma(t);