[Python] How to compute digamma function in complex numbers in order to use this function in...

N= [-7.12843079e+02, -1.39668296e+02, -6.01626070e+01, -3.51688015e+01]
jax.scipy.special.digamma(N)

TypeError: digamma does not accept dtype complex64. Accepted dtypes are subtypes of floating.
​

I'm trying to compute digamma in complex numbers using jax.scipy.special.digamma, however, it still gives me this error even though the doc of this package says it can be complex here is what the doc says:

Parameters: z (array_like) – Real or complex argument.

Any idea how to solve this issue? Or is there an alternative way such as other library or other packages that allows me to use the complex number to compute the digamma function!?

Continue reading...