Say I have a numpy array as follows:
arr = np.array([[[1, 7], [5, 1]], [[5, 6], [6, 7]]])
where each of the innermost sub-arrays is an element. So