Array Processing In Numpy Which Function Flattens