devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Return nan at poles of polygamma, digamma, and zeta if limit is not defined

Browse Source
This commit is contained in:
frgossen
2021-02-19 16:35:11 +00:00
committed by Antonio Sánchez
parent 7f09d3487d
commit 33e0af0130
2 changed files with 22 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
unsupported/Eigen/src/SpecialFunctions/SpecialFunctionsImpl.h
View File

@@ -241,7 +241,7 @@ struct digamma_impl {
Scalar p, q, nz, s, w, y;
bool negative = false;
const Scalar maxnum = NumTraits<Scalar>::infinity();
const Scalar nan = NumTraits<Scalar>::quiet_NaN();
const Scalar m_pi = Scalar(EIGEN_PI);
const Scalar zero = Scalar(0);
@@ -254,7 +254,7 @@ struct digamma_impl {
q = x;
p = numext::floor(q);
if (p == q) {
return maxnum;
return nan;
}
/* Remove the zeros of tan(m_pi x)
* by subtracting the nearest integer from x
@@ -1403,7 +1403,12 @@ struct zeta_impl {
{
if(q == numext::floor(q))
{
return maxnum;
if (x == numext::floor(x) && long(x) % 2 == 0) {
return maxnum;
}
else {
return nan;
}
}
p = x;
r = numext::floor(p);
@@ -1479,11 +1484,11 @@ struct polygamma_impl {
Scalar nplus = n + one;
const Scalar nan = NumTraits<Scalar>::quiet_NaN();
// Check that n is an integer
if (numext::floor(n) != n) {
// Check that n is a non-negative integer
if (numext::floor(n) != n || n < zero) {
return nan;
}
// Just return the digamma function for n = 1
// Just return the digamma function for n = 0
else if (n == zero) {
return digamma_impl<Scalar>::run(x);
}