We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6127350 commit 71d086cCopy full SHA for 71d086c
nn/math_.py
@@ -27,6 +27,11 @@ def gelu(x: LayerRef) -> Layer:
27
return activation(x, activation="gelu")
28
29
30
+def glu(x: LayerRef) -> Layer:
31
+ """GLU https://arxiv.org/abs/1612.08083"""
32
+ return activation(x, activation='glu')
33
+
34
35
def exp(x: LayerRef) -> Layer:
36
"""exp"""
37
return activation(x, activation="exp")
@@ -50,8 +55,3 @@ def sigmoid(x: LayerRef) -> Layer:
50
55
def swish(x: LayerRef) -> Layer:
51
56
"""swish"""
52
57
return activation(x, activation="swish")
53
-
54
-def glu(x: LayerRef) -> Layer:
- """GLU"""
- return activation(x, activation='glu')
0 commit comments