Returns an ndarray of indices of maximum elements in a three-dimensional ndarray a over a given axis.