针对NaN检查Arrayfire阵列

时间:2012-09-25 11:36:40

标签: python arrayfire

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

由于速度问题,使用numpy.nansum / numpy.nan_to_num不是一个选项。

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

2 个答案:

答案 0 :(得分:1)

瓶颈值得研究。他们对numpy.nanxxx函数进行了几次优化,根据我的经验,这些函数比numpy快5倍。

答案 1 :(得分:0)

您可以使用numpy.nansum,它会在数组总和中将nan视为0。

numpy.nansum(arrayfire)