问题描述
在：
a = torch.tensor([[2.4]])
torch.squeeze(a,1)
a.size(),a
出：
(torch.Size([1,1]),tensor([[2.4000]]))
在使用 nn.MSELoss 进行计算时，我发现维度不匹配。
输入有大小 ([1,1]) 和目标 ([])。
函数 reshape 和 squeeze 没有工作。
对于这个令人尴尬的简单问题的解决方案，我将不胜感激。 :]
编辑：有一个简单的错误是未分配 a= 压缩值。谢谢你的回答。
解决方法
函数 torch.squeeze 不会修改输入 a。要么重新分配它：
a = a.squeeze(1)
或使用函数 torch.squeeze_
的就地版本
a.squeeze_(1)
