np.squeeze()

np.squeeze() 是NumPy库中的一个函数，用于从数组中删除单维度的条目。它返回一个在输入数组中删除了尺寸为1的维度的新数组。在上述示例中，我们创建了一个形状为(1, 3, 1)的输入数组arr。然后，我们使用np.squeeze()函数将其变形为(3,)的数组squeezed_arr。通过打印squeezed_arr，我们可以看到被压缩后的数组中不再有尺寸为1的维度。