diff --git a/python/paddle/tensor/search.py b/python/paddle/tensor/search.py
index 9b1dce1df867d97043dbc904831f6f34fbbdff79..3ec79b55b0bbe61daf8820520d4b07e6b10023a0 100755
--- a/python/paddle/tensor/search.py
+++ b/python/paddle/tensor/search.py
@@ -326,7 +326,7 @@ def index_select(x, index, axis=0, name=None):
         check_variable_and_dtype(
             x,
             'x',
-            ['float32', 'float64', 'int32', 'int64'],
+            ['float16', 'float32', 'float64', 'int32', 'int64'],
             'paddle.tensor.search.index_select',
         )
         check_variable_and_dtype(