numpy flatten

latten是numpy.ndarray.flatten的一個函數 

import numpy as np
a = np.zeros((2,3,4))
print(a.shape)
print(a.flatten().shape)
# (2, 3, 4)
# (24,)

可以看到，它把 2*3*4的數組變成了1*24的數組

flatten函數就是將數組摺疊稱一維的輸出

a = [[2, 3, 4], [5, 6, 7]]
a = np.array(a)
print(a)
print(a.size)
print(a.flatten().shape)
print(a.flatten())
#[[2 3 4]
# [5 6 7]]
#6
#(6,)
#[2 3 4 5 6 7]