numpy的squeeze函数和expand_dims函数
squeeze 函数:从数组的形状中删除单维度条目,即把shape中为1的维度去掉
用法:numpy.squeeze(a,axis = None):
- a表示输入的数组;
- axis用于指定需要删除的维度,但是指定的维度必须为单维度,否则将会报错;
- axis的取值可为None 或 int 或tuple of ints, 可选。若axis为空,则删除所有单维度的条目;
- 返...
原创
2019-09-17 19:43:08 ·
3062 阅读 ·
0 评论