Add Binary Cross Entropy loss (#122)

* update BCE added tests for it ... * added binary cross entropy loss to docs * resolving conflicts for merge
2025-09-01 12:49:44 +08:00 · 2023-12-11 16:55:18 +01:00
parent b0cd092b7f
commit f4ddd7dc44
3 changed files with 102 additions and 1 deletions
--- a/docs/src/python/nn.rst
+++ b/docs/src/python/nn.rst
@@ -179,6 +179,7 @@ Loss Functions
   :template: nn-module-template.rst

   losses.cross_entropy
+   losses.binary_cross_entropy
   losses.l1_loss
   losses.mse_loss
   losses.nll_loss