python - numexpr 支持特殊的贝塞尔函数吗？

Question

我需要在大约一千万点处评估 BesselK 函数。我知道scipy.special支持此为 scipy.special.kv(n, x) ，但我想要更快的评估以及内存有效的评估。理想的numexpr会很好，我尝试过 numexpr.evaluate("kv(n, x)")以及 numexpr.evaluate("besselk(n,x)")但两者都不起作用。有谁知道 numexpr 中 BesselK 的确切命令？

Answer 1

numexpr 文档列出了支持的函数，可惜贝塞尔函数不在其中。

http://numexpr.readthedocs.io/projects/NumExpr3/en/latest/user_guide.html

您可以尝试寻找替代实现，例如来自 GSL 或 boost，但坦率地说，我怀疑速度差异是否会是微不足道的，如果有的话。

可能相关的是避免在 python 端的点上循环。要么只使用 scipy.special.kv 的数组值参数，要么放到 Cython 并使用 cython_special 版本并在 Cython 中进行循环。