zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-07-19 23:51:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use compiled function in Sigmoid module (#1116)

Browse Source
This commit is contained in:
Cheng 2024-05-14 21:25:57 +08:00 committed by GitHub
parent 60cb11764e
commit 5be5daa6ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/mlx/nn/layers/activations.py
View File

@ -315,7 +315,7 @@ class GLU(Module):
return glu(x=x, axis=self.axis) return glu(x=x, axis=self.axis)
@_make_activation_module(mx.sigmoid) @_make_activation_module(sigmoid)
class Sigmoid(Module): class Sigmoid(Module):
r"""Applies the sigmoid function, element-wise. r"""Applies the sigmoid function, element-wise.