给定一个（5,2）张量，删除行在第二列中具有重复的行

Question

因此，让我们假设我有这样的张量：

[[0,18],
[1,19],
[2, 3],
[3,19],
[4, 18]]

我需要仅使用TensorFlow删除第二列中包含重复的行。最终的输出应该是：

[[0,18],
[1,19],
[2, 3]]

Answer 1

您应该能够使用tf.math.unsorted_segrent_min和tf.gather：

import tensorflow as tf

x = tf.constant([[0,18],
                 [1,19],
                 [2, 3],
                 [3,19],
                 [4, 18]])


y, idx = tf.unique(x[:, 1])
indices = tf.math.unsorted_segment_min(tf.range(tf.shape(x)[0]), idx, tf.shape(y)[0])
result = tf.gather(x, indices)
print(result)

tf.Tensor(
[[ 0 18]
 [ 1 19]
 [ 2  3]], shape=(3, 2), dtype=int32)

这是对调用tf.unique tf.unique <的解决方案的解决。 /code>：