如何在Tensorflow中移除张量中的重复值？

Question

我最近正在处理单词嵌入，这是我现在面临的问题。我将在张量中具有重复的值。假设我的形状中有以下2D张量（？，5）

输入

[[1 2 4 2 6]
 [7 2 6 1 2]
 [7 2 6 1 2]
 [3 9 3 9 2]
 [3 9 3 9 2]
 [4 9 8 1 6]
 [4 9 8 1 6]]

预期输出

[[1 2 4 2 6]
 [7 2 6 1 2]
 [3 9 3 9 2]
 [4 9 8 1 6]]

我在社区中进行了一些调查。我仍然无法具体解决此问题。有什么方法可以解决tensorflow中的重复问题吗？

Answer 1

添加它们进行设置。 Set将删除重复项，然后转换为列表。

a = list(set(tensor))