diff --git a/python/paddle/tensor/search.py b/python/paddle/tensor/search.py index 3ec79b55b0bbe61daf8820520d4b07e6b10023a0..e16ac89953fb196bbe075012ac8f3317ff1eb7d6 100755 --- a/python/paddle/tensor/search.py +++ b/python/paddle/tensor/search.py @@ -292,7 +292,7 @@ def index_select(x, index, axis=0, name=None): size as the length of ``index``; other dimensions have the same size as in the ``x`` tensor. Args: - x (Tensor): The input Tensor to be operated. The data of ``x`` can be one of float32, float64, int32, int64. + x (Tensor): The input Tensor to be operated. The data of ``x`` can be one of float16, float32, float64, int32, int64. index (Tensor): The 1-D Tensor containing the indices to index. The data type of ``index`` must be int32 or int64. axis (int, optional): The dimension in which we index. Default: if None, the ``axis`` is 0. name(str, optional): For details, please refer to :ref:`api_guide_Name`. Generally, no setting is required. Default: None.