#python #matrix #gpu #cupy
Вопрос:
Рассмотрим следующий код, который умножает две матрицы, заполненные только 0 и 1:
import cupy as cp
import numpy as np
A = cp.array(np.random.randint(2, size=(10000, 10000), dtype='uint8'))
B = cp.array(np.random.randint(2, size=(10000, 10000), dtype='uint8'))
C = cp.matmul(A,B)
Это довольно интенсивная операция, и для ее завершения требуется некоторое время. C снова будет матрица 10000×10000 , заполненная 0 и 1. Давайте теперь скажем, что меня интересует только появление первого 1 в C , есть ли в любом случае, чтобы CuPy остановил вычисления, когда 1 был найден первый?