mlx.nn.losses.nll_loss#

class nll_loss(inputs: array, targets: array, axis: int = -1, reduction: Literal['none', 'mean', 'sum'] = 'none')#

计算负对数似然损失。

参数:

• inputs (array) – 对数空间的预测分布。

• targets (array) – 目标值。

• axis (int, optional) – 分布轴。默认值: -1。

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

返回:

计算出的 NLL 损失。

返回类型:

array