Bias

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	√
Atlas A2系列产品	√
Atlas 200I/500 A2推理产品	√
Atlas推理系列产品	√
Atlas训练系列产品	√

功能说明

Bias 按照 axis、num_axes 和 bias_from_blob 推导 bias 的逻辑广播形状，并计算 y = x + broadcast(bias)。

计算公式如下：

$$y_i=x_i+bia{s}_i$$

broadcast(bias) 的推导规则如下：

• axis < 0 时，先执行 axis += rank(x)。
• bias_from_blob = true 且 num_axes = -1 时，broadcast_shape = [1] * axis + shape(bias)。
• bias_from_blob = true 且 num_axes = 0 时，broadcast_shape = [1] * rank(x)。
• bias_from_blob = true 且 num_axes > 0 时，broadcast_shape = [1] * axis + shape(bias) + [1] * (rank(x) - axis - num_axes)。
• bias_from_blob = false 且 bias 为单元素时，broadcast_shape = [1] * rank(x)。
• bias_from_blob = false 的其它场景，broadcast_shape = [1] * axis + shape(bias) + [1] * (rank(x) - axis - rank(bias))。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待进行Bias计算的输入张量。	FLOAT、FLOAT16、BFLOAT16	ND、NC1HWC0
bias	输入	按属性广播后与x相加的偏置张量。	FLOAT、FLOAT16、BFLOAT16	ND、NC1HWC0
axis	属性	bias 逻辑形状的起始对齐轴，默认值为1。	INT	-
num_axes	属性	bias 覆盖的轴数，默认值为1；支持-1、0和正整数。	INT	-
bias_from_blob	属性	是否按 blob bias 规则推导逻辑广播形状，默认值为true。	BOOL	-
y	输出	输出张量，形状和数据类型跟随x。	FLOAT、FLOAT16、BFLOAT16	ND、NC1HWC0

约束说明

• 支持 ND 和 NC1HWC0 数据格式（4D输入时支持NC1HWC0），以及 FLOAT、FLOAT16、BFLOAT16 同数据类型输入输出；不支持混合数据类型。
• bias 经过属性推导后的每个维度必须为1，或与x的对应维度相同。

调用说明

调用方式	调用样例	说明
图模式调用	test_geir_bias	通过算子IR构图方式调用Bias算子。