python - tensorflow ，删除重复项

Question

我有一个一维张量，例如:

[1,2,2,3,3,4,5]

如何从中获取[1,2,3,4,5]？

我的解决方案是:

我需要一个类似于 contains 的 TensorFlow API。

然后我可以将单独的一个放入 TensorArray 中。

Answer 1

您正在寻找tf.unique。直接引用自documentation :

# tensor 'x' is [1, 1, 2, 4, 4, 4, 7, 8, 8]
y, idx = unique(x)
y ==> [1, 2, 4, 7, 8]
idx ==> [0, 0, 1, 2, 2, 2, 3, 4, 4]