Fix typos in unsqueeze & unsequeeze wrapper

python/paddle/fluid/layers/nn.py

@@ -4485,6 +4485,7 @@ def reshape(x, shape, actual_shape=None, act=None, inplace=True, name=None):
 
     return helper.append_activation(out)
 
+
 def squeeze(x, axes, inplace=False, name=None):
     """
     Remove single-dimensional entries from the shape of a tensor. Takes a 
@@ -4511,7 +4512,7 @@ def squeeze(x, axes, inplace=False, name=None):
     Args:
         x (Variable): The input variable to be squeezed.
         axes (list): List of integers, indicating the dimensions to be squeezed.
-        name (str): Name for this layers.
+        name (str|None): Name for this layer.
 
     Returns:
         Variable: Output squeezed variable.
@@ -4532,6 +4533,7 @@ def squeeze(x, axes, inplace=False, name=None):
 
     return out
 
+
 def unsqueeze(x, axes, inplace=False, name=None):
     """
     Insert single-dimensional entries to the shape of a tensor. Takes one 
@@ -4545,7 +4547,7 @@ def unsqueeze(x, axes, inplace=False, name=None):
     Args:
         x (Variable): The input variable to be unsqueezed.
         axes (list): List of integers, indicating the dimensions to be inserted.
-        name (str): Name for this layers.
+        name (str|None): Name for this layer.
 
     Returns:
         Variable: Output unsqueezed variable.
@@ -4566,6 +4568,7 @@ def unsqueeze(x, axes, inplace=False, name=None):
 
     return out
 
+
 def lod_reset(x, y=None, target_lod=None):
     """
     Set LoD of :attr:`x` to a new one specified by :attr:`y` or

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

@@ -243,7 +243,7 @@ class TestBook(unittest.TestCase):
     def test_sequence_unsqueeze(self):
         program = Program()
         with program_guard(program):
-            x = layers.data(name='x', shape=[8,2], dtype='float32')
+            x = layers.data(name='x', shape=[8, 2], dtype='float32')
             out = layers.unsqueeze(x=x, axes=[1])
             self.assertIsNotNone(out)
         print(str(program))
@@ -252,7 +252,7 @@ class TestBook(unittest.TestCase):
         program = Program()
         with program_guard(program):
             x = layers.data(name='x', shape=[1, 1, 4], dtype='float32')
-            out = layers.squeeze(x=x, axes=[0])
+            out = layers.squeeze(x=x, axes=[2])
             self.assertIsNotNone(out)
         print(str(program))
 
@@ -278,7 +278,6 @@ class TestBook(unittest.TestCase):
             self.assertIsNotNone(out)
         print(str(program))
 
-
     def test_im2sequence(self):
         program = Program()
         with program_guard(program):