BCELoss

    该接口用于创建一个BCELoss的可调用类,用于计算输入 input 和标签 label 之间的二值交叉熵损失值。二值交叉熵损失函数公式如下:

    当 weight 不为空时,公式为:

    当 weight 为空时,公式为:

    BCELoss - 图2

    当 reduction 为 mean 时,最终的输出结果为:

    当 reduction 为 sum 时,最终的输出结果为:

    BCELoss - 图4

    注解