未验证 提交 e62b0c2c 编写于 作者: W wangxiaoning 提交者: GitHub

fix document of index_select (#50168)

上级 d1da885f
......@@ -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.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册