Browse Source

!28748 Fix SoftShrink operator documentation

Merge pull request !28748 from chenweitao_295/SoftShrink_doc_issues
tags/v1.6.0
i-robot Gitee 4 years ago
parent
commit
b7e514f683
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/python/mindspore/ops/operations/nn_ops.py

+ 2
- 2
mindspore/python/mindspore/ops/operations/nn_ops.py View File

@@ -8538,7 +8538,7 @@ class Conv3DTranspose(PrimitiveWithInfer):

class SoftShrink(Primitive):
r"""
Applies the soft shrinkage function element-wise.
Applies the SoftShrink function element-wise.

.. math::
\text{SoftShrink}(x) =
@@ -8549,7 +8549,7 @@ class SoftShrink(Primitive):
\end{cases}

Args:
lambd(Float): the :math:`\lambda` must be no less than zero value for the Softshrink formulation. Default: 0.5.
lambd(Float): the :math:`\lambda` must be no less than zero for the SoftShrink formulation. Default: 0.5.

Inputs:
- **input_x** (Tensor) - The input of SoftShrink with data type of float16 or float32.


Loading…
Cancel
Save