mindspore.ops.UniqueWithPad

class mindspore.ops.UniqueWithPad[源代码]

对输入一维张量中元素去重，返回一维张量中的唯一元素（使用pad_num填充）和相对索引。

基本操作与Unique相同，但UniqueWithPad多了Pad操作。 Unique运算符处理输入张量 x 后所返回的元组（ y ， idx ）， y 与 idx 的shape通常会有差别。因此，为了解决上述情况， UniqueWithPad操作符将用用户指定的 pad_num 填充 y 张量，使其具有与张量 idx 相同的形状。

更多参考详见 mindspore.ops.unique_with_pad()。

输入：

  • x (Tensor) - 需要被去重的Tensor。必须是类型为int32或int64的一维向量。

  • pad_num (int) - 填充值。数据类型为int32或int64。

输出：

Tuple， (y, idx) 。

  • y 是与 x 的shape和数据类型相同的Tensor，包含 x 中去重后的元素，并用 pad_num 填充。

  • idx 为索引Tensor，包含 x 中的元素在 y 中的索引，与 x 的shape相同。

支持平台：

Ascend GPU CPU

样例：

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