减少张量的维度到标量

如何解决减少张量的维度到标量

在：

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)