diff --git a/python/paddle/fluid/layers/ops.py b/python/paddle/fluid/layers/ops.py
index f06c0abaf901950d072cf857696bc7479fb7b52f..03884d618ea8616af0d53539c51618f4a78c5d77 100644
--- a/python/paddle/fluid/layers/ops.py
+++ b/python/paddle/fluid/layers/ops.py
@@ -86,12 +86,11 @@ def uniform_random(shape, dtype='float32', min=-1.0, max=1.0, seed=0):
             result = fluid.layers.uniform_random(shape=[32, 784])
     """
 
-    locals_var = locals().keys()
+    locals_var = locals()
     if not isinstance(dtype, core.VarDesc.VarType):
         dtype = convert_np_dtype_to_dtype_(dtype)
     kwargs = dict()
-    for name in locals_var:
-        val = locals()[name]
+    for name, val in locals_var.items():
         if val is not None:
             kwargs[name] = val
     return _uniform_random_(**kwargs)