diff --git a/python/paddle/distribution.py b/python/paddle/distribution.py
index 918ebce07825139fabe8ddd4c1e266dd04eb7f6d..35204affb3fd168b8bd137d78c3413a08885e2bb 100644
--- a/python/paddle/distribution.py
+++ b/python/paddle/distribution.py
@@ -138,7 +138,7 @@ class Distribution(object):
         convert value's dtype to be consistent with param's dtype.
 
         Args:
-            param (int|float|list|numpy.ndarray|Tensor): low and high in Uniform class, loc and scale in Normal class.
+            param (Tensor): low and high in Uniform class, loc and scale in Normal class.
             value (Tensor): The input tensor.
 
         Returns:
@@ -152,6 +152,7 @@ class Distribution(object):
                 )
                 return core.ops.cast(value, 'in_dtype', value.dtype,
                                      'out_dtype', param.dtype)
+            return value
 
         check_variable_and_dtype(value, 'value', ['float32', 'float64'],
                                  'log_prob')