From e62b0c2cc3332df29c205e71a1e0c497269e394a Mon Sep 17 00:00:00 2001 From: wangxiaoning <71813629+wangxn12138@users.noreply.github.com> Date: Fri, 3 Feb 2023 14:29:09 +0800 Subject: [PATCH] fix document of index_select (#50168) --- python/paddle/tensor/search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/tensor/search.py b/python/paddle/tensor/search.py index 3ec79b55b0..e16ac89953 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. -- GitLab