diff --git a/doc/fluid/api_cn/layers_cn/scatter_nd_add_cn.rst b/doc/fluid/api_cn/layers_cn/scatter_nd_add_cn.rst index 96c2722082e7a67edec3a35679a057adb3d9679b..3a703fb7a4c04f6dabed1257fa1207472ee4c5b6 100644 --- a/doc/fluid/api_cn/layers_cn/scatter_nd_add_cn.rst +++ b/doc/fluid/api_cn/layers_cn/scatter_nd_add_cn.rst @@ -40,7 +40,7 @@ scatter_nd_add 参数: - - **ref** (Variable) - 输入张量,数据类型可以是int32,int64,float32,float64。 + - **ref** (Variable) - 输入张量,数据类型可以是float32,float64。 - **index** (Variable) - 输入的索引张量,数据类型为非负int32或非负int64。它的维度 :code:`index.rank` 必须大于1,并且 :code:`index.shape[-1] <= ref.rank` - **updates** (Variable) - 输入的更新张量,它必须和 :code:`ref` 有相同的数据类型。形状必须是 :code:`index.shape[:-1] + ref.shape[index.shape[-1]:]` 。 - **name** (string) - 该层的名字,默认值为None,表示会自动命名。 diff --git a/doc/fluid/api_cn/layers_cn/scatter_nd_cn.rst b/doc/fluid/api_cn/layers_cn/scatter_nd_cn.rst index 525cac14ab4150b935cf7c2ad259797c0716f024..b0e660b03319f087d7995c9699d117a578c1555e 100644 --- a/doc/fluid/api_cn/layers_cn/scatter_nd_cn.rst +++ b/doc/fluid/api_cn/layers_cn/scatter_nd_cn.rst @@ -9,7 +9,7 @@ scatter_nd 参数: - **index** (Variable) - 输入的索引张量,数据类型为非负int32或非负int64。它的维度 :code:`index.rank` 必须大于1,并且 :code:`index.shape[-1] <= len(shape)` - - **updates** (Variable) - 输入的更新张量。形状必须是 :code:`index.shape[:-1] + shape[index.shape[-1]:]` 。数据类型可以是int32,int64,float32,float64。 + - **updates** (Variable) - 输入的更新张量。形状必须是 :code:`index.shape[:-1] + shape[index.shape[-1]:]` 。数据类型可以是float32,float64。 - **shape** (tuple|list) - 要求输出张量的形状。类型是tuple或者list。 - **name** (string) - 该层的名字,默认值为None,表示会自动命名。