bce_with_logits_loss.hpp File Reference

Binary cross-entropy on raw logits – combined sigmoid + BCE, numerically stable. More...

#include "pulsatrix/device_backend.hpp"
#include "pulsatrix/tensor.hpp"

Include dependency graph for bce_with_logits_loss.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	pulsatrix::BCEWithLogitsLoss
	loss = mean( max(x,0) - x*y + log(1 + exp(-|x|)) ), over all N*k elements of a (N, k) logit tensor x against a target tensor y of the same shape. More...

Namespaces
namespace	pulsatrix

Detailed Description

Binary cross-entropy on raw logits – combined sigmoid + BCE, numerically stable.