Skip to content

Commit b785798

Browse files
author
Yibing Liu
committed
Expose layer's name for sequence pad & unpad
test=develop
1 parent 46e61d8 commit b785798

File tree

2 files changed

+9
-5
lines changed

2 files changed

+9
-5
lines changed

paddle/fluid/API.spec

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,8 +75,8 @@ paddle.fluid.layers.conv2d_transpose ArgSpec(args=['input', 'num_filters', 'outp
7575
paddle.fluid.layers.conv3d_transpose ArgSpec(args=['input', 'num_filters', 'output_size', 'filter_size', 'padding', 'stride', 'dilation', 'groups', 'param_attr', 'bias_attr', 'use_cudnn', 'act', 'name'], varargs=None, keywords=None, defaults=(None, None, 0, 1, 1, None, None, None, True, None, None))
7676
paddle.fluid.layers.sequence_expand ArgSpec(args=['x', 'y', 'ref_level', 'name'], varargs=None, keywords=None, defaults=(-1, None))
7777
paddle.fluid.layers.sequence_expand_as ArgSpec(args=['x', 'y', 'name'], varargs=None, keywords=None, defaults=(None,))
78-
paddle.fluid.layers.sequence_pad ArgSpec(args=['x', 'pad_value', 'maxlen'], varargs=None, keywords=None, defaults=(None,))
79-
paddle.fluid.layers.sequence_unpad ArgSpec(args=['x', 'length'], varargs=None, keywords=None, defaults=None)
78+
paddle.fluid.layers.sequence_pad ArgSpec(args=['x', 'pad_value', 'maxlen', 'name'], varargs=None, keywords=None, defaults=(None, None))
79+
paddle.fluid.layers.sequence_unpad ArgSpec(args=['x', 'length', 'name'], varargs=None, keywords=None, defaults=(None,))
8080
paddle.fluid.layers.lstm_unit ArgSpec(args=['x_t', 'hidden_t_prev', 'cell_t_prev', 'forget_bias', 'param_attr', 'bias_attr', 'name'], varargs=None, keywords=None, defaults=(0.0, None, None, None))
8181
paddle.fluid.layers.reduce_sum ArgSpec(args=['input', 'dim', 'keep_dim', 'name'], varargs=None, keywords=None, defaults=(None, False, None))
8282
paddle.fluid.layers.reduce_mean ArgSpec(args=['input', 'dim', 'keep_dim', 'name'], varargs=None, keywords=None, defaults=(None, False, None))

python/paddle/fluid/layers/nn.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2793,7 +2793,7 @@ def sequence_expand_as(x, y, name=None):
27932793

27942794

27952795
@templatedoc()
2796-
def sequence_pad(x, pad_value, maxlen=None):
2796+
def sequence_pad(x, pad_value, maxlen=None, name=None):
27972797
"""
27982798
${comment}
27992799
@@ -2807,7 +2807,9 @@ def sequence_pad(x, pad_value, maxlen=None):
28072807
None or any positive int. When it is None, all sequences will be
28082808
padded up to the length of the longest one among them; when it a
28092809
certain positive value, it must be greater than the length of the
2810-
longest original sequence."
2810+
longest original sequence.
2811+
name(str|None): A name for this layer(optional). If set None, the layer
2812+
will be named automatically.
28112813
28122814
Returns:
28132815
Variable: The padded sequence batch and the original lengths before
@@ -2844,7 +2846,7 @@ def sequence_pad(x, pad_value, maxlen=None):
28442846
return out, length
28452847

28462848

2847-
def sequence_unpad(x, length):
2849+
def sequence_unpad(x, length, name=None):
28482850
"""
28492851
Sequence Unpad Layer
28502852
@@ -2876,6 +2878,8 @@ def sequence_unpad(x, length):
28762878
equal length.
28772879
length(Variable): The Variable that specifies the actual ength of
28782880
sequences after unpadding.
2881+
name(str|None): A name for this layer(optional). If set None, the layer
2882+
will be named automatically.
28792883
28802884
Returns:
28812885
Variable: The Variable contains the unpadded sequences.

0 commit comments

Comments
 (0)