forked from mindspore-Ecosystem/mindspore
!49650 [bugfix] update soft shrink
Merge pull request !49650 from shaojunsong/softshrink2
This commit is contained in:
commit
5426703477
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -5,4 +5,4 @@ mindspore.ops.SoftShrink
|
|||
|
||||
Soft Shrink激活函数。
|
||||
|
||||
更多参考详见 :func:`mindspore.ops.soft_shrink`。
|
||||
更多参考详见 :func:`mindspore.ops.softshrink`。
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -458,6 +458,7 @@ from .nn_func import (
|
|||
selu,
|
||||
softmax,
|
||||
softmin,
|
||||
softshrink,
|
||||
pdist,
|
||||
pad,
|
||||
prelu,
|
||||
|
|
|
@ -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',
|
||||
|
|
|
@ -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``
|
||||
|
|
Loading…
Reference in New Issue