Pytorch_torch.nn.MSELoss
均方损失函数作用主要是求预测实例与真实实例之间的loss
loss(xi,yi)=(xi−yi)2
- 函数需要输入两个tensor,类型统一设置为float,否则会报错,也可以在全局设置torch.set_default_tensor_type(torch.FloatTensor),也可以在计算时转换
loss=torch.nn.MSELoss()
c=torch.tensor([[1,2],[3,4]])
d=torch.tensor([[5,6],[7,8]])
loss(c.float(),d.float())
- 单个tensor正常求即可,多个求其平均loss。
item()方法
如果tensor只有一个元素,调用item会将tensor转为python中scalars;如果不是一个元素,则会报错。