pytorch BCEWithLogitsLoss equivalence

[wangjiawen2013]

Hi,
pytorch has BCEWithLogitsLoss, is there the burn equivalence ?

[laggui]

Yep, you can use the BinaryCrossEntropyLoss with logits:

let loss = BinaryCrossEntropyLossConfig::new().with_logits(true).init(&device);