xx = np.arange(27).reshape(3, 3, -1)
for x in xx:
y = x.reshape(1, -1)
print(x)
print(y)
print('-----------------------------')
结果:
[[0 1 2]
[3 4 5]
[6 7 8]]
[[0 1 2 3 4 5 6 7 8]]
-----------------------------
[[ 9 10 11]
[12 13 14]
[15 16 17]]
[[ 9 10 11 12 13 14 15 16 17]]
-----------------------------
[[18 19 20]
[21 22 23]
[24 25 26]]
[[18 19 20 21 22 23 24 25 26]]
-----------------------------