2.5K Star 8.4K Fork 4.7K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_addmv.rst 1.68 KB
一键复制 编辑 原始数据 按行查看 历史
changzherui 提交于 2023-06-15 20:36 . modify api param name

mindspore.ops.addmv

.. py:function:: mindspore.ops.addmv(input, mat, vec, *, beta=1, alpha=1)

    `mat` 和 `vec` 相乘,且将输入向量 `input` 加到最终结果中。

    如果 `mat` 是一个大小为 :math:`(N, M)` Tensor, `vec` 一个大小为 :math:`M` 的一维Tensor,那么 `input` 必须是可广播的,且
    带有一个大小为 :math:`N` 的一维Tensor。这种情况下 `out` 是一个大小为 :math:`N` 的一维Tensor。

    可选值 `bata` 和 `alpha` 分别是 `mat` 和 `vec` 矩阵向量的乘积和附加Tensor `input` 的扩展因子。如果 `beta` 为0,那么 `input` 将被忽略。

    .. math::
        output = β input + α (mat @ vec)

    参数:
        - **input** (Tensor) - 被加的向量,Tensor的shape大小为 :math:`(N,)`。
        - **mat** (Tensor) - 第一个需要相乘的Tensor,shape大小为 :math:`(N, M)` 。
        - **vec** (Tensor) - 第二个需要相乘的Tensor,shape大小为 :math:`(M,)` 。

    关键字参数:
        - **beta** (scalar[int, float, bool], 可选) - `input` (β)的乘数。 `beta` 必须是int或float或bool类型,默认值: ``1`` 。
        - **alpha** (scalar[int, float, bool], 可选) - `mat` @ `vec` (α)的乘数。 `alpha` 必须是int或float或bool类型,默认值: ``1`` 。

    返回:
        Tensor,shape大小为 :math:`(N,)` ,数据类型与 `input` 相同。

    异常:
        - **TypeError** - `mat` 、 `vec` 、 `input` 不是Tensor。
        - **TypeError** - `mat` 、 `vec` 的数据类型不一致。
        - **ValueError** - 如果 `mat` 不是一个二维Tensor。
        - **ValueError** - 如果 `vec` 不是一个一维Tensor。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.q1

搜索帮助