mlx.nn.losses.mse_loss

class mse_loss(predictions: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'mean')

计算均方误差损失。

参数：

• predictions (array) – 预测值。

• targets (array) – 目标值。

• reduction (str, optional) – 指定应用于输出的归约方法：'none' | 'mean' | 'sum'。默认值：'mean'。

返回值：

计算出的均方误差损失。

返回类型：

array