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]