Loss Functions | 实现 | 名称 | 适用场景 |
---|---|---|---|
nn.L1Loss | 创建一个标准来测量输入中每个元素x和目标y之间的平均绝对误差(MAE) | 平均绝对值误差损失 | 回归 |
nn.MSELoss | 创建一个标准,用于测量输入中每个元素x和目标y之间的均方误差(L2范数的平方)。 | 均方误差损失 | 回归 |
nn.CrossEntropyLoss | 该准则计算输入logits与目标之间的交叉熵损失。 | 交叉熵损失 | 多分类 |
nn.CTCLoss | 连接主义时间分类损失。 | ||
nn.NLLLoss | 负对数似然损失。 | 负对数似然函数损失 | 多分类 |
nn.PoissonNLLLoss | 目标泊松分布的负对数似然损失。 | ||
nn.GaussianNLLLoss | 高斯负对数似然损失。 | ||
nn.KLDivLoss | Kullback-Leibler散度损失。 | KL 散度损失 | 回归 |
nn.BCELoss | 创建一个标准,测量 |