mindspore.ops.bias_add

mindspore.ops.bias_add(input_x, bias)[源代码]

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

参数:
  • input_x (Tensor) - 输入Tensor。shape可以有2~5个维度。数据类型应为float16或float32。

  • bias (Tensor) - 偏置Tensor,shape为 \((C)\)。C必须与 input_x 的通道维度C相同,数据类型应为float16或float32。

返回:

Tensor,shape和数据类型与 input_x 相同。

异常:
  • TypeError - input_xbias 不是Tensor。

  • TypeError - input_xbias 的数据类型既不是float16也不是float32。

  • TypeError - input_xbias 的数据类型不一致。

  • TypeError - input_x 的维度不在[2, 5]范围内。

支持平台:

Ascend GPU CPU

样例:

>>> input_x = Tensor(np.arange(6).reshape((2, 3)), mindspore.float32)
>>> bias = Tensor(np.random.random(3).reshape((3)), mindspore.float32)
>>> output = ops.bias_add(input_x, bias)
>>> print(output.shape)
(2, 3)