El método flatten

El método numpy.ndarray.flatten devuelve una copia del array colapsado a una única dimensión. Veamos un ejemplo sencillo:

a = np.array([[1, 2, 3], [4, 5, 6]])
a

El método flatten

Si ejecutamos el método en cuestión, obtenemos lo siguiente:

a.flatten()

array([1, 2, 3, 4, 5, 6])

NumPy ofrece diferentes criterios para este "colapso": por filas, columnas, etc. Revisa la documentación para más información.