Сумма большого numpy массива в кусках - PullRequest
1 голос
/ 29 февраля 2020

У меня большой 3D-набор данных h5py frames Я хочу суммировать по первому измерению. Он не помещается все в память, поэтому я должен сделать:

sh = frames.shape[0]

temp_1 = frames[:sh//2,...].sum(0)
temp_2 = frames[sh//2:,...].sum(0)

tot_img = temp_1 + temp_2

Есть ли способ оптимизировать это так, чтобы он вычислялся быстрее?

...