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()
。- 支持平台:
Ascend
GPU
CPU
样例:
>>> x = Tensor(np.array([1, 1, 5, 5, 4, 4, 3, 3, 2, 2,]), mindspore.int32) >>> pad_num = 8 >>> output = ops.UniqueWithPad()(x, pad_num) >>> print(output) (Tensor(shape=[10], dtype=Int32, value= [1, 5, 4, 3, 2, 8, 8, 8, 8, 8]), Tensor(shape=[10], dtype=Int32, value= [0, 0, 1, 1, 2, 2, 3, 3, 4, 4]))