Modify mask of multidimensional numpy array

Question

I am working with multidimensional numpy arrays and masks.
I have an array a of shape (10000, 20, 4, 4) and create a mask by searching that a