pytorch中的squeeze()、unsqueeze()和unsqueeze_()

1.unsqueeze()函数

从tensor（torch）或者数组中，在指定维度上添加维度为1的维度（axis=0, axis=1, axis=2, …）

2.squeeze()函数

从tensor（torch）或者数组中，在指定维度上删除维度为1的维度（axis=0, axis=1, axis=2, …）

3.unsqueeze_()

unsqueeze_和unsqueeze实现一样的功能, 区别在于unsqueeze_是in_place操作,即unsqueeze不会对使用unsqueeze的tensor进行改变,想要获取unsqueeze后的值必须赋予个新值,unsqueeze_则会对自己改变

对于Tensor操作的几个函数，包括.reshape(), .view(), .permute()。涉及到维度的变化，主要是用来进行矩阵乘法运算，一定要注意维度！！！