本文共 533 字，大约阅读时间需要 1 分钟。

def bceloss(output, target):    positive_prob = F.logsigmoid(output)    negative_prob = F.logsigmoid(-output)    loss = -positive_prob*target-negative_prob*(1-target)    loss = loss.mean()    return loss    class bceloss(nn.Module):    def __init__(self):        super(bceloss, self).__init__()            def forward(self, pred, target):        positive_prob = F.logsigmoid(pred)        negative_prob = F.logsigmoid(-1 * pred)        loss = -positive_prob*target-negative_prob*(1-target)        loss = loss.mean()        return loss

转载地址：http://tqyen.baihongyu.com/