cross_entropy_loss.hpp File Reference

Softmax + negative log-likelihood classification loss. More...

#include <cstdint>
#include "pulsatrix/device_backend.hpp"
#include "pulsatrix/tensor.hpp"

Include dependency graph for cross_entropy_loss.hpp:

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

Go to the source code of this file.

Classes
class	pulsatrix::CrossEntropyLoss
	loss = -log(softmax(logits)[target_class]), combined for numerical stability (subtract the max logit before exponentiating) rather than computing softmax and log separately. More...

Namespaces
namespace	pulsatrix

Detailed Description

Softmax + negative log-likelihood classification loss.