2.3K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_bias_add.rst 904 Bytes
一键复制 编辑 原始数据 按行查看 历史
luojianing 提交于 2023-09-12 10:09 . fix docs issues

mindspore.ops.bias_add

.. py:function:: mindspore.ops.bias_add(input_x, bias)

    返回输入Tensor `input_x` 与偏置Tensor `bias` 之和。相加前会把偏置Tensor广播成与输入Tensor的shape一致。

    参数:
        - **input_x** (Tensor) - 输入Tensor。shape可以有2~5个维度。支持数据类型:

          - Ascend/CPU: all Number type。
          - GPU: float16、float32、int8。

        - **bias** (Tensor) - 偏置Tensor,shape为 :math:`(C)`。C必须与 `input_x` 的通道维度C相同。其数据类型与 `input_x` 一致。

    返回:
        Tensor,shape和数据类型与 `input_x` 相同。

    异常:
        - **TypeError** - `input_x` 或  `bias` 不是Tensor。
        - **TypeError** - `input_x` 与  `bias` 的数据类型不一致。
        - **TypeError** - `input_x` 的维度不在[2, 5]范围内。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助