Свернуть измерение трехмерного массива в двумерное с помощью logic_or и Numpy

Question

У меня есть трехмерный массив двумерных масок (x, y) (для контекста маски были созданы видео), и я хочу свернуть маски в одно измерение x, y. У меня есть около 600 масок в массиве. Есть хорошая функция numpy, которая может сделать это?

#setup
mask = np.zeros([3,4,5], dtype=int)
mask[0,2,2] = 1
mask[0,3,1] = 1
mask[1,0,2] = 1
mask[1,1,3] = 1
mask[2,0,1] = 1
mask[2,1,2] = 1
#collapse operation
res = np.zeros_like(mask[0])
for i in mask:
    res += i
res