Skip to content

Commit bf3c994

Browse files
authored
add unique_consecutive_cn (PaddlePaddle#3689)
* add unique_consecutive_cn * add unique_consecutive op * add unique_consecutive default value * add unique_consecutive copyfrom * add unique_consecutive axis default value
1 parent 1a43d24 commit bf3c994

File tree

2 files changed

+26
-0
lines changed

2 files changed

+26
-0
lines changed

docs/api/paddle/Overview_cn.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -274,6 +274,7 @@ tensor元素操作相关(如:转置,reshape等)
274274
" :ref:`paddle.transpose <cn_api_fluid_layers_transpose>` ", "根据perm对输入的多维Tensor进行数据重排"
275275
" :ref:`paddle.unbind <cn_api_paddle_tensor_unbind>` ", "将输入Tensor按照指定的维度分割成多个子Tensor"
276276
" :ref:`paddle.unique <cn_api_tensor_cn_unique>` ", "返回Tensor按升序排序后的独有元素"
277+
" :ref:`paddle.unique_consecutive <cn_api_tensor_cn_unique_consecutive>` ", "返回无连续重复元素的Tensor"
277278
" :ref:`paddle.unsqueeze <cn_api_paddle_tensor_unsqueeze>` ", "该OP向输入Tensor的Shape中一个或多个位置(axis)插入尺寸为1的维度"
278279
" :ref:`paddle.unstack <cn_api_fluid_layers_unstack>` ", "该OP将单个dim为 D 的Tensor沿 axis 轴unpack为 num 个dim为 (D-1) 的Tensor"
279280

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
.. _cn_api_tensor_cn_unique_consecutive:
2+
3+
unique_consecutive
4+
-------------------------------
5+
6+
.. py:function:: paddle.unique_consecutive(x, return_inverse=False, return_counts=False, axis=None, dtype="int64", name=None)
7+
8+
将Tensor中连续重复的元素进行去重,返回连续不重复的Tensor。
9+
10+
参数:
11+
- **x** (Tensor) - 输入的 `Tensor` ,数据类型为:float32、float64、int32、int64。
12+
- **return_inverse** (bool, 可选) - 如果为True,则还返回输入Tensor的元素对应在连续不重复元素中的索引,该索引可用于重构输入Tensor。默认:False.
13+
- **return_counts** (bool, 可选) - 如果为True,则还返回每个连续不重复元素在输入Tensor中的个数。默认:False.
14+
- **axis** (int, 可选) - 指定选取连续不重复元素的轴。默认值为None,将输入平铺为1-D的Tensor后再选取连续不重复元素。默认:None.
15+
- **dtype** (np.dtype|str, 可选) - 用于设置 `inverse` 或者 `counts` 的类型,应该为int32或者int64。默认:int64.
16+
- **name** (str,可选)- 具体用法请参见 :ref:`api_guide_Name` ,一般无需设置,默认值为None。
17+
18+
返回:
19+
- **out** (Tensor) - 连续不重复元素构成的Tensor,数据类型与输入一致。
20+
- **inverse** (Tensor, 可选) - 输入Tensor的元素对应在连续不重复元素中的索引,仅在 `return_inverse` 为True时返回。
21+
- **counts** (Tensor, 可选) - 每个连续不重复元素在输入Tensor中的个数,仅在 `return_counts` 为True时返回。
22+
23+
**代码示例**:
24+
25+
COPY-FROM: paddle.unique_consecutive

0 commit comments

Comments
 (0)