2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_prelu.rst 1.21 KB
一键复制 编辑 原始数据 按行查看 历史
俞涵 提交于 2024-09-23 14:38 . modify format

mindspore.ops.prelu

.. py:function:: mindspore.ops.prelu(input, 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` 是通道权重。

    PReLU函数图:

    .. image:: ../images/PReLU2.png
        :align: center

    .. note::
        通道数是输入的第二个维度值。当输入的维度小于2时,则没有通道维度并且通道数等于1。

    参数:
        - **input** (Tensor) - 激活函数的输入Tensor。
        - **weight** (Tensor) - 权重Tensor。它的大小是1或者输入Tensor `input` 的通道数。

    返回:
        Tensor,其shape和数据类型与 `input` 相同。
        有关详细信息,请参考 :class:`mindspore.mint.nn.PReLU` 。

    异常:
        - **TypeError** - `input` 或 `weight` 不是Tensor。
        - **ValueError** - `weight` 不是0-D或1-DTensor。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助