diff --git a/python/paddle/fluid/layers/nn.py b/python/paddle/fluid/layers/nn.py
index 8dced4bbfcbfb924b0ea69dceeeb11de3f2a821f..9656dcf94f14ad9250bb7e79c1330c9bdd44d9d6 100644
--- a/python/paddle/fluid/layers/nn.py
+++ b/python/paddle/fluid/layers/nn.py
@@ -3295,7 +3295,7 @@ def autoincreased_step_counter(counter_name=None, begin=1, step=1):
     return counter
 
 
-def lod_reset(x, y, target_lod=None):
+def lod_reset(x, y=None, target_lod=None):
     """
     LoD Reset Operator. Set LoD of **x** to a new one specified by **y** or
     **target_lod**. When **y** provided, **y.lod** would be considered as target