Add Gaussian NLL loss function (#477)

* Add Gaussian NLL loss function --------- Co-authored-by: Awni Hannun <awni@apple.com>
2025-09-26 15:58:14 +08:00 · 2024-01-18 22:44:44 +08:00
parent 9c111f176d
commit d1fef34138
3 changed files with 126 additions and 17 deletions
--- a/docs/src/python/nn/losses.rst
+++ b/docs/src/python/nn/losses.rst
@@ -12,6 +12,7 @@ Loss Functions
   binary_cross_entropy
   cosine_similarity_loss
   cross_entropy
+   gaussian_nll_loss
   hinge_loss
   huber_loss
   kl_div_loss