np.squeeze()与np.squeeze()函数详解

1. np.squeeze()

语法：numpy.squeeze(a,axis = None)

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

• 作用：从数组的形状中删除单维度条目，即把shape中为1的维度去掉
  具体用法参考：
  https://blog.csdn.net/zenghaitao0128/article/details/78512715

2. np.unsqueeze()

语法：torch.unsqueeze(input, dim, out=None)

• 作用：扩展维度
  返回一个新的张量，对输入的既定位置插入维度 1
• 注意： 返回张量与输入张量共享内存，所以改变其中一个的内容会改变另一个。
  如果dim为负，则将会被转化dim+input.dim()+1

1）tensor (Tensor) – 输入张量
2）dim (int) – 插入维度的索引
3）out (Tensor, optional) – 结果张量

具体用法参考：
https://zhuanlan.zhihu.com/p/86763381