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