使用logical_or和Numpy将3D数组的维折叠为2D

Question

我有一个二维遮罩（x，y）的3维数组（对于上下文，这些遮罩是由视频生成的），我想将这些遮罩折叠成一个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