Python で Arrayfire を使用していますが、入力配列に NaN が含まれており、NAN を合計として返すため、af.sum() 関数を使用できません。
numpy.nansum/numpy.nan_to_num の使用は、速度の問題のためオプションではありません。
これらの NaN を arrayfire で浮動小数点ゼロに変換する方法が必要なだけです。
ボトルネックは調べる価値があります。彼らは numpy.nanxxx 関数に対していくつかの最適化を実行しました。これにより、私の経験では、numpy よりも約 5 倍速くなります。
配列 sum で 0 として扱うnumpy.nansumを使用できます。nan
numpy.nansum(arrayfire)