Flatten层将多维的输入一维化
二维输入:
a=array([[1,2],
[3,4],
[5,6]])
#shape[3,2]
a.flatten() #默认按行的方向降维
#Out:[1, 2, 3, 4, 5, 6])
a.flatten('F') #按列降维
#Out:[1, 3, 5, 2, 4, 6])
a.flatten('A') #按行降维
#Out:[1, 2, 3, 4, 5, 6]
三维输入:
a=array([[[1,2],
[3,4]],
[[5,6],
[7,8]]])
#shape[2,2,2]
a.flatten() #默认按行的方向降维
#Out:[1, 2, 3, 4, 5, 6, 7, 8])