如何使用cupy循环numpy ndarray？真的会缩短执行时间吗？

Question

我有两个大小相同的numpy ndarray [512 X 512]。我想根据其他数组值更改一个数组。但是执行时间确实很长。因此，我想将numpy ndarray更改为cupy ndarray并希望在GPU中执行。可能吗？是会缩短执行时间吗？这是我当前的代码。

for n,val in enumerate(array_A.flat):
    if val < 200:
        Array_B.flat[n] = -1000

Answer 1

如果使用NumPy的内置函数对选项进行矢量化处理，则性能将大大提高：

Database name to filter

这会将操作推到较低的级别，避免在Python中执行循环。