在Python中，squeeze()函数用于去除数组中的单维度条目。具体来说，它会删除shape为1的维度，即将维度大小为1的维度去掉。
例如，假设有一个形状为(1, 5)的数组，使用squeeze()函数后，它将变成形状为(5,)的数组。
import numpy as np
arr = np.array([[1, 2, 3, 4, 5]])
print(arr.shape) # 输出 (1, 5)
arr_squeezed = np.squeeze(arr)
print(arr_squeezed.shape) # 输出 (5,)
squeeze()函数还可以指定去除某个特定的维度，例如axis=0表示去除第一个维度。
import numpy as np
arr = np.array([[[1, 2, 3, 4, 5]]])
print(arr.shape) # 输出 (1, 1, 5)
arr_squeezed = np.squeeze(arr, axis=0)
print(arr_squeezed.shape) # 输出 (1, 5)
