numpy筆記—np.squeeze用法

import numpy as np x = np.array([[[0], [1], [2]]]) print(x.shape)

d = np.squeeze(x) # 從數組的形狀中刪除單維條目，即把shape中為1的維度去掉 print(d.shape)