forked from mindspore-Ecosystem/mindspore
!38823 silu bugfix
Merge pull request !38823 from Henry Shi/ops_bugfix_silu_sofmin
This commit is contained in:
commit
6618a4bfcd
|
@ -592,6 +592,8 @@ class SiLU(Cell):
|
|||
>>> x = Tensor(np.array([-1, 2, -3, 2, -1]), mindspore.float16)
|
||||
>>> silu = nn.SiLU()
|
||||
>>> output = silu(x)
|
||||
>>> print(output)
|
||||
[-0.269 1.762 -0.1423 1.762 -0.269]
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
|
|
Loading…
Reference in New Issue