zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Softshrink mapping + op (#552)

Browse Source 
* Added Softshrink mapping + op

* formatting

* docs + nits in docstring

---------

Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
Andre Slavescu
2024-01-30 15:56:28 -05:00
committed by GitHub
parent 3f7aba8498
commit d3a9005454
5 changed files with 70 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/src/python/nn/functions.rst
View File

@@ -19,5 +19,6 @@ simple functions.
prelu
relu
selu
softshrink
silu
step