np.squeeze?

np.squeeze()是一个NumPy函数，用于从数组的形状中删除单维度条目。如果数组的形状中有一个或多个维度大小为1，则可以使用此函数将其删除。这对于减少数组的维数非常有用。

例如，如果有一个形状为(1, 3, 1, 5)的数组，使用np.squeeze()函数将其转换为形状为(3, 5)的数组。