|
|
|
@ -1649,7 +1649,7 @@ class BiasAdd(PrimitiveWithInfer):
|
|
|
|
|
- **input_x** (Tensor) - The input tensor. The shape can be 2-4 dimensions.
|
|
|
|
|
- **bias** (Tensor) - The bias tensor, with shape :math:`(C)`.
|
|
|
|
|
- **data_format** (str) - The format of input and output data. It should be 'NHWC' or 'NCHW',\
|
|
|
|
|
default is 'NCHW'.
|
|
|
|
|
default is 'NCHW'.
|
|
|
|
|
The shape of `bias` must be the same as `input_x` in the second dimension.
|
|
|
|
|
|
|
|
|
|
Outputs:
|
|
|
|
|