Skip to content

Commit eb1aeb1

Browse files
authored
Merge pull request #13538 from baiyfbupt/softshrink
Remove kwargs in softshrink
2 parents d635854 + 3986242 commit eb1aeb1

File tree

3 files changed

+10
-2
lines changed

3 files changed

+10
-2
lines changed

paddle/fluid/API.spec

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -265,12 +265,12 @@ paddle.fluid.layers.sum ArgSpec(args=[], varargs='args', keywords='kwargs', defa
265265
paddle.fluid.layers.slice ArgSpec(args=[], varargs='args', keywords='kwargs', defaults=None)
266266
paddle.fluid.layers.shape ArgSpec(args=[], varargs='args', keywords='kwargs', defaults=None)
267267
paddle.fluid.layers.maxout ArgSpec(args=[], varargs='args', keywords='kwargs', defaults=None)
268-
paddle.fluid.layers.softshrink ArgSpec(args=[], varargs='args', keywords='kwargs', defaults=None)
269268
paddle.fluid.layers.sigmoid ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=(None,))
270269
paddle.fluid.layers.logsigmoid ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=(None,))
271270
paddle.fluid.layers.exp ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=(None,))
272271
paddle.fluid.layers.tanh ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=(None,))
273272
paddle.fluid.layers.tanh_shrink ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=(None,))
273+
paddle.fluid.layers.softshrink ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=(None,))
274274
paddle.fluid.layers.sqrt ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=(None,))
275275
paddle.fluid.layers.abs ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=(None,))
276276
paddle.fluid.layers.ceil ArgSpec(args=['x', 'name'], varargs=None, keywords=None, defaults=(None,))

python/paddle/fluid/layers/ops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
'exp',
2222
'tanh',
2323
'tanh_shrink',
24+
'softshrink',
2425
'sqrt',
2526
'abs',
2627
'ceil',
@@ -52,7 +53,6 @@
5253
'slice',
5354
'shape',
5455
'maxout',
55-
'softshrink',
5656
]
5757

5858
for _OP in set(__all__):

python/paddle/fluid/tests/unittests/test_layers.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -758,6 +758,14 @@ def test_expand(self):
758758
out = layers.expand(x, [1, 2])
759759
print(str(program))
760760

761+
def test_softshrink(self):
762+
program = Program()
763+
with program_guard(program):
764+
input = layers.data(name="input", shape=[16], dtype="float32")
765+
out = layers.softshrink(input, name='softshrink')
766+
self.assertIsNotNone(out)
767+
print(str(program))
768+
761769

762770
if __name__ == '__main__':
763771
unittest.main()

0 commit comments

Comments
 (0)