2.6K Star 8.6K Fork 4.9K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
mindspore.mint.nn.functional.linear.rst 1.21 KB
一键复制 编辑 原始数据 按行查看 历史
shilishan 提交于 2024-10-30 21:11 . pynative customize dense

mindspore.mint.nn.functional.linear

.. py:function:: mindspore.mint.nn.functional.linear(input, weight, bias=None)

    对输入 `input` 应用全连接操作。全连接定义为:

    .. math::
        output = input * weight^{T} + bias

    .. warning::
        - 这是一个实验性API,后续可能修改或删除。
        - 在PYNATIVE模式下,如果 `bias` 不是1D, `input` 不可以大于6D。

    参数:
        - **input** (Tensor) - 输入Tensor,shape是 :math:`(*, in\_channels)`,其中 :math:`*` 表示任意的附加维度。
        - **weight** (Tensor) - 输入Tensor的权重,shape是 :math:`(out\_channels, in\_channels)` 或 :math:`(in\_channels)`。
        - **bias** (Tensor,可选) - 添加在输出结果的偏差,shape是 :math:`(out\_channels)` 或 :math:`()`。默认值:``None`` ,偏差为0。

    返回:
        输出结果,shape由 `input` 和 `weight` 的shape决定。

    异常:
        - **TypeError** - `input` 不是Tensor。
        - **TypeError** - `weight` 不是Tensor。
        - **TypeError** - `bias` 不是Tensor。
        - **RuntimeError** - 在PYNATIVE模式下, `bias` 不是1D且 `input` 大于6D。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385