mindspore.ops.unique

查看源文件
mindspore.ops.unique(input)[源代码]

对输入tensor中元素去重。

警告

这是一个实验性API,后续可能修改或删除。

参数:
  • input (Tensor) - 输入tensor。

返回:

由两个tensor组成的tuple(output, indices)。

  • output (Tensor) - 去重后的输出。

  • indices (Tensor) - 输入tensor的元素在 output 中的索引。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> x = mindspore.tensor([1, 2, 5, 2], mindspore.int32)
>>> output, indices = mindspore.ops.unique(x)
>>> print(output)
[1 2 5]
>>> print(indices)
[0 1 2 1]