Using a target size (torch.Size([100])) that is different to the input size (torch.Size([100, 1])) is deprecated. Please ensure they have the same size.
解决办法
1.squeeze(1)和squeeze(-1)作用:
两者的效果一样,都是给张量tensor降维,但不是啥张量都可以用这两个函数来降维,它只能降维一种情况下张量的维度。就是我的张量tensor是一个n1维度的张量,例如:张量[[1], [2], [3]]是一个31维的,调用这两个函数后效果如图:
2.squeeze(0)的作用:
当张量是一个1n维度的张量时,例如:张量[[1, 2, 3]]是一个13维的,调用这个函数后的效果图如下
3.unsqueeze(1)和unsqueeze(-1)作用:
和squeeze(1)和squeeze(-1)是反着的,如果我就是一个一维的张量,我调用这个函数,我就变成上面第1节中的原本的样子。
例如:张量[1.2, -5.6, 9, 0.004],调用这两个函数后的效果为: