From 47ca29caefec99b7230ccea7a12e29161f96c17d Mon Sep 17 00:00:00 2001 From: root Date: Mon, 2 Dec 2019 02:33:41 +0000 Subject: [PATCH] Update the unique_with_counts document, add the unique op could support the GPU --- doc/fluid/api_cn/layers_cn/unique_with_counts_cn.rst | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/doc/fluid/api_cn/layers_cn/unique_with_counts_cn.rst b/doc/fluid/api_cn/layers_cn/unique_with_counts_cn.rst index b0f27acf1..e4fc5ca75 100644 --- a/doc/fluid/api_cn/layers_cn/unique_with_counts_cn.rst +++ b/doc/fluid/api_cn/layers_cn/unique_with_counts_cn.rst @@ -6,7 +6,7 @@ unique_with_counts .. py:function:: paddle.fluid.layers.unique_with_counts(x, dtype='int32') 该OP对输入Tensor元素进行去重，获取去重后结果Tensor，同时获取去重后结果在原始输入中的计数Tensor以及在原始输入中的索引Tensor。 -注:该OP仅支持 **CPU** ，同时仅支持 **Tensor** +注:该OP仅支持 **Tensor** 参数： - **x** (Variable) – 数据shape为 :math:`[N]` 的一维Tensor，数据类型为 float32，float64，int32，int64。 -- GitLab