Skip to content

Commit 64ca3d1

Browse files
authored
Add bias_attr in sequence_conv_pool API. (#14553)
1 parent e52d90a commit 64ca3d1

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

paddle/fluid/API.spec

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,7 @@ paddle.fluid.transpiler.RoundRobin.dispatch ArgSpec(args=['self', 'varlist'], va
342342
paddle.fluid.transpiler.RoundRobin.reset ArgSpec(args=['self'], varargs=None, keywords=None, defaults=None)
343343
paddle.fluid.transpiler.DistributeTranspilerConfig.__init__
344344
paddle.fluid.nets.simple_img_conv_pool ArgSpec(args=['input', 'num_filters', 'filter_size', 'pool_size', 'pool_stride', 'pool_padding', 'pool_type', 'global_pooling', 'conv_stride', 'conv_padding', 'conv_dilation', 'conv_groups', 'param_attr', 'bias_attr', 'act', 'use_cudnn'], varargs=None, keywords=None, defaults=(0, 'max', False, 1, 0, 1, 1, None, None, None, True))
345-
paddle.fluid.nets.sequence_conv_pool ArgSpec(args=['input', 'num_filters', 'filter_size', 'param_attr', 'act', 'pool_type'], varargs=None, keywords=None, defaults=(None, 'sigmoid', 'max'))
345+
paddle.fluid.nets.sequence_conv_pool ArgSpec(args=['input', 'num_filters', 'filter_size', 'param_attr', 'act', 'pool_type', 'bias_attr'], varargs=None, keywords=None, defaults=(None, 'sigmoid', 'max', None))
346346
paddle.fluid.nets.glu ArgSpec(args=['input', 'dim'], varargs=None, keywords=None, defaults=(-1,))
347347
paddle.fluid.nets.scaled_dot_product_attention ArgSpec(args=['queries', 'keys', 'values', 'num_heads', 'dropout_rate'], varargs=None, keywords=None, defaults=(1, 0.0))
348348
paddle.fluid.nets.img_conv_group ArgSpec(args=['input', 'conv_num_filter', 'pool_size', 'conv_padding', 'conv_filter_size', 'conv_act', 'param_attr', 'conv_with_batchnorm', 'conv_batchnorm_drop_rate', 'pool_stride', 'pool_type', 'use_cudnn'], varargs=None, keywords=None, defaults=(1, 3, None, None, False, 0.0, 1, 'max', True))

python/paddle/fluid/nets.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -250,7 +250,8 @@ def sequence_conv_pool(input,
250250
filter_size,
251251
param_attr=None,
252252
act="sigmoid",
253-
pool_type="max"):
253+
pool_type="max",
254+
bias_attr=None):
254255
"""
255256
The sequence_conv_pool is composed with Sequence Convolution and Pooling.
256257
@@ -266,6 +267,11 @@ def sequence_conv_pool(input,
266267
pool_type (str): Pooling type can be :math:`max` for max-pooling, :math:`average` for
267268
average-pooling, :math:`sum` for sum-pooling, :math:`sqrt` for sqrt-pooling.
268269
Default :math:`max`.
270+
bias_attr (ParamAttr|bool|None): The parameter attribute for the bias of sequence_conv.
271+
If it is set to False, no bias will be added to the output units.
272+
If it is set to None or one attribute of ParamAttr, sequence_conv
273+
will create ParamAttr as bias_attr. If the Initializer of the bias_attr
274+
is not set, the bias is initialized zero. Default: None.
269275
270276
Return:
271277
Variable: The final result after Sequence Convolution and Pooling.
@@ -289,6 +295,7 @@ def sequence_conv_pool(input,
289295
num_filters=num_filters,
290296
filter_size=filter_size,
291297
param_attr=param_attr,
298+
bias_attr=bias_attr,
292299
act=act)
293300

294301
pool_out = layers.sequence_pool(input=conv_out, pool_type=pool_type)

0 commit comments

Comments
 (0)