损失函数MSELoss
时间: 2023-09-27 15:04:33 浏览: 104
Pytorch 的损失函数Loss function使用详解
MSELoss是均方损失函数的英文名字,它的公式如下所示:(xi-yi)的平方,其中loss、x、y的维度是一样的,可以是向量或者矩阵,i是下标。很多的损失函数都有size_average和reduce两个布尔类型的参数。因为一般损失函数都是直接计算batch的数据,所以返回的loss结果都是维度为(batch_size,)的向量。
阅读全文