mse_loss

    该OP用于计算预测值和目标值的均方差误差。

    对于预测值input和目标值label,公式为:

    当 reduction 设置为 'none' 时,

    当 reduction 设置为 时,

    当 reduction 设置为 'sum' 时,

    , 输入 input 和标签 间的 mse loss 损失。