numpy.nditer.remove_axis
numpy.nditer.reset
When the “multi_index” flag was specified, this removes it, allowing the internal iteration structure to be optimized further.