我在 Python 上使用 Arrayfire,但我不能使用 af.sum() 函数,因为我的输入数组中有 NaN,它会将 NAN 作为总和返回。
由于速度问题,不能使用 numpy.nansum/numpy.nan_to_num 。
我只需要一种方法将这些 NaN 转换为 arrayfire 中的浮点零。
瓶颈值得研究。他们对 numpy.nanxxx 函数进行了多项优化,根据我的经验,这使其比 numpy 快 5 倍左右。
您可以使用numpy.nansum,它nan
在数组总和中视为 0。
numpy.nansum(arrayfire)