pytorch torch.nn.MSELoss(size_average=True)(均方误差损失函数)

Posted Dontla

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch torch.nn.MSELoss(size_average=True)(均方误差损失函数)相关的知识,希望对你有一定的参考价值。

class torch.nn.MSELoss(size_average=True)[source]

创建一个衡量输入x(模型预测输出)和目标y之间均方误差标准。

  • x 和 y 可以是任意形状,每个包含n个元素。

  • 对n个元素对应的差值的绝对值求和,得出来的结果除以n。

  • 如果在创建MSELoss实例的时候在构造函数中传入size_average=False,那么求出来的平方和将不会除以n

参考文章:https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch-nn/#class-torchnnmselosssize_averagetruesource

以上是关于pytorch torch.nn.MSELoss(size_average=True)(均方误差损失函数)的主要内容,如果未能解决你的问题,请参考以下文章

PyTorch学习笔记:nn.MSELoss——MSE损失

Pytorch MSELoss

Pytorch MSELoss

神经网络架构pytorch-MSELoss损失函数

利用pytorch建立神经网络

pytorch p12 简化版本气温预测