Additoinal losses (#336)

* cosine similarity loss --------- Co-authored-by: Awni Hannun <awni@apple.com> * Docstring nits
2025-09-11 22:44:38 +08:00 · 2024-01-09 07:01:13 +09:00
parent 432ee5650b
commit 0b8aeddac6
3 changed files with 74 additions and 12 deletions
--- a/docs/src/python/nn/losses.rst
+++ b/docs/src/python/nn/losses.rst
@@ -19,4 +19,5 @@ Loss Functions
   triplet_loss
   hinge_loss
   huber_loss
-   log_cosh_loss
+   log_cosh_loss
+   cosine_similarity_loss