fix GLU Activation docstrings

This commit is contained in:
Laurent 2024-03-29 17:50:05 +00:00 committed by Laureηt
parent f26b6ee00a
commit 04e59bf3d9

View file

@ -140,10 +140,10 @@ class GLU(Activation):
Example:
```py
glu = fl.GLU()
tensor = torch.tensor([[-1.0, 0.0, 1.0]])
glu = fl.GLU(fl.ReLU())
tensor = torch.tensor([[1.0, 0.0, -1.0, 1.0]])
output = glu(tensor)
assert torch.allclose(output, torch.tensor([0.0, 0.0]))
```
"""