BiasAdd#
Add bias to channel dimension of input. This is a special Add
with
bias restricted to be 1-D. Broadcasting is supported.
\[\dst(n,c,h,w) = \src(n,c,h,w) + \bias(c)\]
Operation Attributes#
|
Description |
Value Type |
|
|
---|---|---|---|---|
Controls
how to
interpret
the shape
of |
string |
|
Optional |
Execution Arguments#
The inputs and outputs must be provided according to the below index order when constructing an operation.
Inputs#
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
|
Required |
1 |
|
Required |
@note bias
is a 1D tensor to be added to src
tensor. The size
should be the same as size of channel dimension of src
tensor.
Outputs#
Index |
Argument Name |
Required or Optional |
---|---|---|
0 |
|
Required |
Supported Data Types#
BiasAdd operation supports the following data type combinations.
Src |
Bias |
Dst |
---|---|---|
f32 |
f32 |
f32 |
bf16 |
bf16 |
bf16 |
f16 |
f16 |
f16 |