o iV@s8ddlZddlmZddlmZddlmZddZdS)N)_digamma)_gamma)_zetacCsRt||\}}d|dt|dt|d|}t|dkt||S)a Polygamma function n. Args: n (cupy.ndarray): The order of the derivative of `psi`. x (cupy.ndarray): Where to evaluate the polygamma function. Returns: cupy.ndarray: The result. .. seealso:: :data:`scipy.special.polygamma` gg?r) cupybroadcast_arraysrgammarzetawhererdigamma)nxfac2r[/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupyx/scipy/special/_polygamma.py polygammas *r)rcupyx.scipy.specialrrrrrrrrs