mindspore.ops.bias_add
- 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为 \((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]范围内。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore >>> import numpy as np >>> from mindspore import Tensor, ops >>> 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)