Softshrink mapping + op (#552)

* Added Softshrink mapping + op * formatting * docs + nits in docstring --------- Co-authored-by: Awni Hannun <awni@apple.com>
2025-09-26 15:58:14 +08:00 · 2024-01-30 15:56:28 -05:00
parent 3f7aba8498
commit d3a9005454
5 changed files with 70 additions and 21 deletions
--- a/docs/src/python/nn/functions.rst
+++ b/docs/src/python/nn/functions.rst
@@ -19,5 +19,6 @@ simple functions.
   prelu
   relu
   selu
+   softshrink
   silu
   step
--- a/docs/src/python/nn/layers.rst
+++ b/docs/src/python/nn/layers.rst
@@ -33,5 +33,6 @@ Layers
   Sequential
   SiLU
   SinusoidalPositionalEncoding
+   Softshrink
   Step
   Transformer