mindspore.ops.unique

mindspore.ops.unique(input)[源代码]

对输入Tensor中元素去重，并返回一个索引Tensor，包含输入Tensor中的元素在输出Tensor中的索引。

y 与 idx 的shape通常会有差别，因为 y 会将元素去重，而 idx 的shape与 input 一致。若需要 idx 与 y 的shape相同，请参考 mindspore.ops.UniqueWithPad 算子。

警告

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

参数：

input (Tensor) - 需要被去重的Tensor。shape： \((N,*)\) ，其中 \(*\) 表示任意数量的附加维度。

返回：

Tuple， (y, idx) 。 y 是与 input 数据类型相同的Tensor，包含 input 中去重后的元素。 idx 为索引Tensor，包含 input 中的元素在 y 中的索引，与 input 的shape相同。

异常：

TypeError - input 不是Tensor。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, nn
>>> from mindspore import ops
>>> x = Tensor(np.array([1, 2, 5, 2]), mindspore.int32)
>>> output = ops.unique(x)
>>> print(output)
(Tensor(shape=[3], dtype=Int32, value= [1, 2, 5]), Tensor(shape=[4], dtype=Int32, value= [0, 1, 2, 1]))
>>> y = output[0]
>>> print(y)
[1 2 5]
>>> idx = output[1]
>>> print(idx)
[0 1 2 1]