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

2024-05-07 • 问答

我最近正在处理单词嵌入，这是我现在面临的问题。我将在张量中具有重复的值。假设我的形状中有以下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]]

我在社区中做过一些调查。我仍然无法具体解决此问题。有什么方法可以解决张量流中的重复问题吗？

您可以使用tf.unique(x)

u,idx = tf.unique(x)
print(u)
# [[1 2 4 2 6]
# [7 2 6 1 2]
# [3 9 3 9 2]
# [4 9 8 1 6]]

本文链接：https://www.f2er.com/3054727.html