From e3c8731002a66d125429feb7f3b3a4dd763562d7 Mon Sep 17 00:00:00 2001 From: ShenLiang <1422485404@qq.com> Date: Wed, 27 Nov 2019 21:45:27 +0800 Subject: [PATCH] fix the type of index, test=develop (#1617) --- doc/fluid/api_cn/layers_cn/gather_nd_cn.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doc/fluid/api_cn/layers_cn/gather_nd_cn.rst b/doc/fluid/api_cn/layers_cn/gather_nd_cn.rst index 65261fc83..d43cbf3cf 100644 --- a/doc/fluid/api_cn/layers_cn/gather_nd_cn.rst +++ b/doc/fluid/api_cn/layers_cn/gather_nd_cn.rst @@ -51,7 +51,7 @@ gather_nd 参数: - **input** (Variable) - 输入张量,数据类型可以是int32,int64,float32,float64, bool。 - - **index** (int) - 输入的索引张量,数据类型为非负int32或非负int64。它的维度 :code:`index.rank` 必须大于1,并且 :code:`index.shape[-1] <= input.rank` 。 + - **index** (Variable) - 输入的索引张量,数据类型为非负int32或非负int64。它的维度 :code:`index.rank` 必须大于1,并且 :code:`index.shape[-1] <= input.rank` 。 - **name** (string) - 该层的名字,默认值为None,表示会自动命名。 返回:shape为index.shape[:-1] + input.shape[index.shape[-1]:]的Tensor|LoDTensor,数据类型与 :code:`input` 一致。 -- GitLab