未验证 提交 1635c02b 编写于 作者: Z Zeng Jinle 提交者: GitHub

Fix scatter_nd_add doc (#35542)

* fix scatter_nd_add doc, test=document_fix

* update
test=document_fix
上级 2b105211
...@@ -8612,7 +8612,7 @@ def scatter_nd_add(ref, index, updates, name=None): ...@@ -8612,7 +8612,7 @@ def scatter_nd_add(ref, index, updates, name=None):
output = [[67, 19], [-16, -27]] output = [[67, 19], [-16, -27]]
Args: Args:
ref (Variable): The ref input. Its dtype should be float32, float64. ref (Variable): The ref input. Its dtype should be int32, int64, float32, float64.
index (Variable): The index input with rank > 1 and index.shape[-1] <= ref.rank. index (Variable): The index input with rank > 1 and index.shape[-1] <= ref.rank.
Its dtype should be int32 or int64 as it is used as indexes. Its dtype should be int32 or int64 as it is used as indexes.
updates (Variable): The updated value of scatter_nd_add op, and it must have the same dtype updates (Variable): The updated value of scatter_nd_add op, and it must have the same dtype
......
...@@ -1348,7 +1348,7 @@ def scatter_nd_add(x, index, updates, name=None): ...@@ -1348,7 +1348,7 @@ def scatter_nd_add(x, index, updates, name=None):
output = [[67, 19], [-16, -27]] output = [[67, 19], [-16, -27]]
Args: Args:
x (Tensor): The x input. Its dtype should be float32, float64. x (Tensor): The x input. Its dtype should be int32, int64, float32, float64.
index (Tensor): The index input with ndim > 1 and index.shape[-1] <= x.ndim. index (Tensor): The index input with ndim > 1 and index.shape[-1] <= x.ndim.
Its dtype should be int32 or int64 as it is used as indexes. Its dtype should be int32 or int64 as it is used as indexes.
updates (Tensor): The updated value of scatter_nd_add op, and it must have the same dtype updates (Tensor): The updated value of scatter_nd_add op, and it must have the same dtype
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册