forked from huawei/mindspore2022
33 lines
881 B
ReStructuredText
33 lines
881 B
ReStructuredText
mindspore.nn.LeakyReLU
|
||
=======================
|
||
|
||
.. py:class:: mindspore.nn.LeakyReLU(alpha=0.2)
|
||
|
||
Leaky ReLU激活函数。
|
||
|
||
该激活函数定义如下：
|
||
|
||
.. math::
|
||
\text{leaky_relu}(x) = \begin{cases}x, &\text{if } x \geq 0; \cr
|
||
{\alpha} * x, &\text{otherwise.}\end{cases}
|
||
|
||
其中，:math:`\alpha` 表示 `alpha` 参数。
|
||
|
||
更多细节详见 `Rectifier Nonlinearities Improve Neural Network Acoustic Models <https://ai.stanford.edu/~amaas/papers/relu_hybrid_icml2013_final.pdf>`_。
|
||
|
||
**参数：**
|
||
|
||
**alpha** (`Union[int, float]`) – x<0时激活函数的斜率，默认值：0.2。
|
||
|
||
**输入：**
|
||
|
||
**x** （Tensor） - 计算LeakyReLU的任意维度的Tensor。
|
||
|
||
**输出：**
|
||
|
||
Tensor，数据类型和shape与 `x` 相同。
|
||
|
||
**异常：**
|
||
|
||
**TypeError** - `alpha` 不是浮点数或整数。
|