mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix

This commit is contained in:
chenweitao_295 2022-01-10 09:57:03 +08:00
parent 3b00621bd3
commit 815b4669e5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/python/mindspore/ops/operations/nn_ops.py
View File

@ -8535,7 +8535,7 @@ class Conv3DTranspose(PrimitiveWithInfer):
class SoftShrink(Primitive): class SoftShrink(Primitive):
r""" r"""
Applies the soft shrinkage function element-wise. Applies the SoftShrink function element-wise.
.. math:: .. math::
\text{SoftShrink}(x) = \text{SoftShrink}(x) =
@ -8546,7 +8546,7 @@ class SoftShrink(Primitive):
\end{cases} \end{cases}
Args: 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: Inputs:
- **input_x** (Tensor) - The input of SoftShrink with data type of float16 or float32. - **input_x** (Tensor) - The input of SoftShrink with data type of float16 or float32.