[机器学习] Pytorch19种损失函数理解[上]— L1、MSE、CrossEntropy、KL、BCE、BCEWithLogits loss function

损失函数通过torch.nn包实现。

文章目录

    • 1 基本用法
    • 2 损失函数(前6种)
      • 2-1 L1范数损失 —— L1_Loss
      • 2-2 均方误差损失 —— MSELoss
      • 2-3 交叉熵损失 CrossEntropyLoss
      • 2-4 KL 散度损失 KLDivLoss
      • 2-5 二元交叉熵损失 BCELoss
      • 2-6 BCEWithLogitsLoss

1 基本用法

criterion = LossCriterion() #构造函数有自己的参数
loss = criterion(x

你可能感兴趣的:(机器学习)