tensorflow，删除重复

Question

我有一个像张力一样的张量：

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

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

我的解决方案是：

我需要一个类似于contains的张量流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]