python - 根据 NaN 检查 Arrayfire 阵列

Question

我在 Python 上使用 Arrayfire，我不能使用 af.sum() 函数，因为我的输入数组中有 NaN，它会返回 NAN 作为总和。

由于速度问题，不能使用 numpy.nansum/numpy.nan_to_num。

我只需要一种方法将这些 NaN 转换为 arrayfire 中的浮点零。

Answer 1

瓶颈值得研究。他们对 numpy.nanxxx 函数进行了多项优化，根据我的经验，这使得它比 numpy 快 5 倍左右。