!49650 [bugfix] update soft shrink

Merge pull request !49650 from shaojunsong/softshrink2
This commit is contained in:
i-robot 2023-03-03 08:56:17 +00:00 committed by Gitee
commit 5426703477
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
8 changed files with 7 additions and 7 deletions

View File

@ -102,9 +102,9 @@ MindSpore中 `mindspore.ops` 接口与上一版本相比,新增、删除和支
mindspore.ops.selu
mindspore.ops.sigmoid
mindspore.ops.silu
mindspore.ops.soft_shrink
mindspore.ops.softmax
mindspore.ops.softmin
mindspore.ops.softshrink
mindspore.ops.softsign
mindspore.ops.tanh
mindspore.ops.threshold

View File

@ -264,7 +264,6 @@ mindspore.Tensor
mindspore.Tensor.sinc
mindspore.Tensor.size
mindspore.Tensor.slogdet
mindspore.Tensor.soft_shrink
mindspore.Tensor.split
mindspore.Tensor.sqrt
mindspore.Tensor.square

View File

@ -5,4 +5,4 @@ mindspore.ops.SoftShrink
Soft Shrink激活函数。
更多参考详见 :func:`mindspore.ops.soft_shrink`。
更多参考详见 :func:`mindspore.ops.softshrink`。

View File

@ -270,7 +270,6 @@
mindspore.Tensor.sinc
mindspore.Tensor.size
mindspore.Tensor.slogdet
mindspore.Tensor.soft_shrink
mindspore.Tensor.split
mindspore.Tensor.sqrt
mindspore.Tensor.square

View File

@ -102,9 +102,9 @@ Activation Functions
mindspore.ops.selu
mindspore.ops.sigmoid
mindspore.ops.silu
mindspore.ops.soft_shrink
mindspore.ops.softmax
mindspore.ops.softmin
mindspore.ops.softshrink
mindspore.ops.softsign
mindspore.ops.tanh
mindspore.ops.threshold

View File

@ -458,6 +458,7 @@ from .nn_func import (
selu,
softmax,
softmin,
softshrink,
pdist,
pad,
prelu,

View File

@ -2346,7 +2346,7 @@ def softmin(x, axis=-1):
def softshrink(x, lambd=0.5):
r"""
Applies the SoftShrink function element-wise.
Applies the Softshrink function element-wise.
.. math::
\text{SoftShrink}(x) =
@ -6134,6 +6134,7 @@ __all__ = [
'hardtanh',
'huber_loss',
'softsign',
'softshrink',
'selu',
'silu',
'softmax',

View File

@ -8298,7 +8298,7 @@ class SoftShrink(Primitive):
r"""
Applies the SoftShrink function element-wise.
Refer to :func:`mindspore.ops.soft_shrink` for more details.
Refer to :func:`mindspore.ops.softshrink` for more details.
Supported Platforms:
``Ascend`` ``GPU`` ``CPU``