提交 e3c87310 编写于 作者: S ShenLiang 提交者: Yi Liu

fix the type of index, test=develop (#1617)

上级 422168d9
...@@ -51,7 +51,7 @@ gather_nd ...@@ -51,7 +51,7 @@ gather_nd
参数: 参数:
- **input** (Variable) - 输入张量,数据类型可以是int32,int64,float32,float64, bool。 - **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,表示会自动命名。 - **name** (string) - 该层的名字,默认值为None,表示会自动命名。
返回:shape为index.shape[:-1] + input.shape[index.shape[-1]:]的Tensor|LoDTensor,数据类型与 :code:`input` 一致。 返回:shape为index.shape[:-1] + input.shape[index.shape[-1]:]的Tensor|LoDTensor,数据类型与 :code:`input` 一致。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册