去掉numpy数组的冗余维度

使用np.squeeze去掉长度为一的维度

自记

示例

原数据如下,第三个维度为一

gen_sig.shape   # (1000, 1024, 1)

使用np.squeeze函数后

gen_sig = np.squeeze(gen_sig)   # (1000, 1024)

你可能感兴趣的:(去掉numpy数组的冗余维度)