Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!38823 silu bugfix 

Merge pull request !38823 from Henry Shi/ops_bugfix_silu_sofmin
This commit is contained in:
i-robot 2022-07-26 09:47:06 +00:00 committed by Gitee
parent d24f4e65e9 74c1cf2ca4
commit 6618a4bfcd
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/nn/layer/activation.py
View File

@ -592,6 +592,8 @@ class SiLU(Cell):
>>> x = Tensor(np.array([-1, 2, -3, 2, -1]), mindspore.float16) >>> x = Tensor(np.array([-1, 2, -3, 2, -1]), mindspore.float16)
>>> silu = nn.SiLU() >>> silu = nn.SiLU()
>>> output = silu(x) >>> output = silu(x)
>>> print(output)
[-0.269 1.762 -0.1423 1.762 -0.269]
""" """
def __init__(self): def __init__(self):