diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index 5a14b9fdc7b6d963f77eefc29476bc332f3938df..bc9f182d95e3b728fbc0866e1c79f5508d3a04aa 100755
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -6309,7 +6309,7 @@ def unsqueeze(input, axes, name=None):
         if isinstance(axes, int):
             axes = [axes]
         elif isinstance(axes, Variable):
-            axes = [axes.numpy().item(0)]
+            axes = axes.numpy().tolist()
         elif isinstance(axes, (list, tuple)):
             axes = [
                 item.numpy().item(0) if isinstance(item, Variable) else item
diff --git a/python/paddle/fluid/tests/unittests/test_unsqueeze_op.py b/python/paddle/fluid/tests/unittests/test_unsqueeze_op.py
index 6f713172f1b29d0df8eed212ab1b148d00d7d45e..1975e4306026ee459aa585c47afa74fce6a6aeed 100644
--- a/python/paddle/fluid/tests/unittests/test_unsqueeze_op.py
+++ b/python/paddle/fluid/tests/unittests/test_unsqueeze_op.py
@@ -167,8 +167,9 @@ class API_TestDyUnsqueezeAxisTensor(unittest.TestCase):
         with fluid.dygraph.guard():
             input1 = np.random.random([5, 10]).astype("int32")
             out1 = np.expand_dims(input1, axis=1)
+            out1 = np.expand_dims(out1, axis=2)
             input = fluid.dygraph.to_variable(input1)
-            output = paddle.unsqueeze(input, axis=paddle.to_tensor([1]))
+            output = paddle.unsqueeze(input, axis=paddle.to_tensor([1, 2]))
             out_np = output.numpy()
             self.assertTrue(np.array_equal(out1, out_np))
             self.assertEqual(out1.shape, out_np.shape)