API(create_tensor) error message enhancement (#23571)

5 years ago · f39899a43d
parent b465bb0de7
commit f39899a43d
1 changed files with 4 additions and 0 deletions
--- a/python/paddle/fluid/layers/tensor.py
+++ b/python/paddle/fluid/layers/tensor.py
@ -57,6 +57,10 @@ def create_tensor(dtype, name=None, persistable=False):
          import paddle.fluid as fluid
          tensor = fluid.layers.create_tensor(dtype='float32')
    """
+    check_dtype(dtype, 'dtype', [
+        'bool', 'float16', 'float32', 'float64', 'int8', 'int32', 'int32',
+        'int64'
+    ], 'create_tensor')
    helper = LayerHelper("create_tensor", **locals())
    return helper.create_variable(
        name=helper.name, dtype=dtype, persistable=persistable)