diff --git a/docs/api/api_python/mindspore.ops.rst b/docs/api/api_python/mindspore.ops.rst index 362f5d5c509..9e9a8ec8059 100644 --- a/docs/api/api_python/mindspore.ops.rst +++ b/docs/api/api_python/mindspore.ops.rst @@ -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 diff --git a/docs/api/api_python/mindspore/mindspore.Tensor.rst b/docs/api/api_python/mindspore/mindspore.Tensor.rst index 3f8a942ed1d..cae9bd93791 100644 --- a/docs/api/api_python/mindspore/mindspore.Tensor.rst +++ b/docs/api/api_python/mindspore/mindspore.Tensor.rst @@ -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 diff --git a/docs/api/api_python/ops/mindspore.ops.SoftShrink.rst b/docs/api/api_python/ops/mindspore.ops.SoftShrink.rst index ef669e229b0..3549bb94ff0 100644 --- a/docs/api/api_python/ops/mindspore.ops.SoftShrink.rst +++ b/docs/api/api_python/ops/mindspore.ops.SoftShrink.rst @@ -5,4 +5,4 @@ mindspore.ops.SoftShrink Soft Shrink激活函数。 - 更多参考详见 :func:`mindspore.ops.soft_shrink`。 \ No newline at end of file + 更多参考详见 :func:`mindspore.ops.softshrink`。 diff --git a/docs/api/api_python_en/Tensor_list.rst b/docs/api/api_python_en/Tensor_list.rst index 412a694bb37..ebf4d8b1407 100644 --- a/docs/api/api_python_en/Tensor_list.rst +++ b/docs/api/api_python_en/Tensor_list.rst @@ -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 diff --git a/docs/api/api_python_en/mindspore.ops.rst b/docs/api/api_python_en/mindspore.ops.rst index bb0fa7021c3..810698136fc 100644 --- a/docs/api/api_python_en/mindspore.ops.rst +++ b/docs/api/api_python_en/mindspore.ops.rst @@ -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 diff --git a/mindspore/python/mindspore/ops/function/__init__.py b/mindspore/python/mindspore/ops/function/__init__.py index 3b13d4ea34b..979dfd19d46 100644 --- a/mindspore/python/mindspore/ops/function/__init__.py +++ b/mindspore/python/mindspore/ops/function/__init__.py @@ -458,6 +458,7 @@ from .nn_func import ( selu, softmax, softmin, + softshrink, pdist, pad, prelu, diff --git a/mindspore/python/mindspore/ops/function/nn_func.py b/mindspore/python/mindspore/ops/function/nn_func.py index 77c316a9d86..c3c116da4f5 100644 --- a/mindspore/python/mindspore/ops/function/nn_func.py +++ b/mindspore/python/mindspore/ops/function/nn_func.py @@ -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', diff --git a/mindspore/python/mindspore/ops/operations/nn_ops.py b/mindspore/python/mindspore/ops/operations/nn_ops.py index 927cc9b00a9..875fa3393f9 100644 --- a/mindspore/python/mindspore/ops/operations/nn_ops.py +++ b/mindspore/python/mindspore/ops/operations/nn_ops.py @@ -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``