/
envoy.istio
/
mindspore
Обзор
Документация
Войти
/
envoy.istio
/
mindspore
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
docs/api/api_python/ops/mindspore.ops.func_prelu.rst
35 строк
1 KB
huan
add image links in files 2.3
05 дек 2023, 05:50
05 дек 2023, 05:50
a5741c0
Код
Авторство
О чём код?
mindspore.ops.prelu =================== .. py:function:: mindspore.ops.prelu(x, weight) 带参数的线性修正单元激活函数（Parametric Rectified Linear Unit activation function）。 `Delving Deep into Rectifiers:Surpassing Human-Level Performance on ImageNet Classification <https://arxiv.org/abs/1502.01852>`_ 描述了PReLU激活函数。定义如下： .. math:: prelu(x_i)= \max(0, x_i) + \min(0, w * x_i), 其中 :math:`x_i` 是输入的一个通道的一个元素，`w` 是通道权重。 .. note:: Ascend不支持标量和一维Tensor的输入。 PReLU函数图： .. image:: ../images/PReLU.png :align: center 参数： - **x** (Tensor) - 激活函数的输入Tensor。数据类型为float16或float32。shape为 :math:`(N, *)` ，其中 :math:`*` 表示任意的附加维度。 - **weight** (Tensor) - 权重Tensor。数据类型为float16或float32。 `weight` 只可以是Tensor，长度与输入x的通道数C相同。在GPU设备上，当输入为标量时，shape为 :math:`(1,)` 。 返回： Tensor，其shape和数据类型与 `x` 相同。 有关详细信息，请参考 :class:`mindspore.nn.PReLU` 。 异常： - **TypeError** - `x` 或 `weight` 的数据类型既不是float16也不是float32。 - **TypeError** - `x` 或 `weight` 不是Tensor。 - **ValueError** - `x` 是Ascend上的0-D或1-D Tensor。 - **ValueError** - `weight` 不是1-D Tensor。