I have the following line with associated indexing
arr = np.array([[[1, 1, 0], [0, 2, 2]], [[1, 2, 0], [0, 1, 1]]]) lab = np.array([[[0, 0, 0], [0, 1, 1]], [[