bias_add¶
dragon.nn.
bias_add
(
inputs,
data_format='NCHW',
inplace=False,
**kwargs
)[source]¶Add the bias across channels to input.
Examples:
x = dragon.ones((1, 2, 3, 3)) bias = dragon.ones((2,)) print(dragon.nn.bias_add([x, bias]))
- Parameters:
- inputs (Sequence[dragon.Tensor]) – The
x
andbias
. - data_format (str, optional, default='NCHW') –
'NCHW'
or'NHWC'
. - inplace (bool, optional, default=False) – Call in-place or return a new tensor.
- inputs (Sequence[dragon.Tensor]) – The
- Returns:
dragon.Tensor – The output tensor.