|
|
|
@@ -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. |
|
|
|
|