Flatten Function In Numpy