1
Amy 原创整理｜资料核验：2026-10-31
【官方API事实】np.squeeze会移除长度为1的轴；不指定axis时会移除所有这类轴。指定的轴若长度不是1，会报ValueError。返回值是原数组自身或它的视图，所有轴都移除时得到的是零维数组。
【教学示例】数据形状为(1,5,1)，其中第一轴表示批次，最后一轴是待移除的单通道。np.squeeze(x, axis=2)得到(1,5)；直接np.squeeze(x)却会得到(5,)，批次这一层也消失了。
【Amy建议】在数据交接处写下预期shape，并明确允许删除哪个轴。尤其测试批次大小为1时，默认squeeze可能让代码表现与多样本批次不同。
官方来源：
https://numpy.org/doc/stable/reference/generated/numpy.squeeze.html