numpy.squeeze()的用法

 

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

x1 = np.squeeze(x)  # 從數組的形狀中刪除單維條目，即把shape中為1的維度去掉
print(x1)  # [0 1 2]
print(x1.shape)  # (3,)