forked from mindspore-Ecosystem/mindspore
26 lines
852 B
ReStructuredText
26 lines
852 B
ReStructuredText
mindspore.ops.hardshrink
|
||
========================
|
||
|
||
.. py:function:: mindspore.ops.hardshrink(x, lambd=0.5)
|
||
|
||
Hard Shrink激活函数，按输入元素计算输出。公式定义如下：
|
||
|
||
.. math::
|
||
\text{HardShrink}(x) =
|
||
\begin{cases}
|
||
x, & \text{ if } x > \lambda \\
|
||
x, & \text{ if } x < -\lambda \\
|
||
0, & \text{ otherwise }
|
||
\end{cases}
|
||
|
||
参数：
|
||
- **x** (Tensor) - Hard Shrink的输入，数据类型为float16或float32。
|
||
- **lambd** (float) - Hard Shrink公式定义的阈值 :math:`\lambda` 。默认值：0.5。
|
||
|
||
返回：
|
||
Tensor，shape和数据类型与输入 `x` 相同。
|
||
|
||
异常：
|
||
- **TypeError** - `lambd` 不是float。
|
||
- **TypeError** - `x` 不是Tensor。
|
||
- **TypeError** - `x` 的dtype既不是float16也不是float32。 |