3

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

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

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

4

2 回答 2

1

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

于 2012-09-25T13:43:25.730 回答
0

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

numpy.nansum(arrayfire)
于 2012-09-25T11:40:10.997 回答