TorchLean API

NN.Proofs.Autograd.Tape.Nodes.Losses

Loss Tape Nodes #

Differentiable loss nodes used by training and verification examples: MSE, cross entropy with one-hot targets, negative log likelihood, BCE-with-logits, and KL divergence.