numpy.squeeze

https://blog.csdn.net/tracy_leaf/article/details/79297121

 

squeeze 函数&＃xff1a;从数组的形状中删除单维度条目&＃xff0c;即把shape中为1的维度去掉

用法&＃xff1a;numpy.squeeze(a,axis &＃61; None)

1&＃xff09;a表示输入的数组&＃xff1b;2&＃xff09;axis用于指定需要删除的维度&＃xff0c;但是指定的维度必须为单维度&＃xff0c;否则将会报错&＃xff1b;3&＃xff09;axis的取值可为None 或 int 或 tuple of ints, 可选。若axis为空&＃xff0c;则删除所有单维度的条目&＃xff1b;4&＃xff09;返回值&＃xff1a;数组5) 不会修改原数组&＃xff1b;


>>> a &＃61; e.reshape(1,1,10)
>>> a
array([[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]]])
>>> np.squeeze(a)
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])