0

同じサイズ[512 X 512]のnumpy ndarrayが2つあります。ある配列を他の配列値に従って変更したい。しかし、実行時間は本当に長いです。そこで、numpy ndarray を cupy ndarray に変更して GPU で実行したい。出来ますか?実行時間はカットされますか?これが私の現在のコードの一部です。

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

1 に答える 1