python - 根据 tensorflow 中的两列对张量进行排序

Question

是否可以在 Tensorflow 中根据两列中的值对张量进行排序？

例如，假设我有以下张量。

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

我愿意喜欢先根据第一列排序，然后根据第二列排序。排序后如下所示。

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

有什么方法可以使用 tensorflow 实现这一点？我能够根据单个列进行排序。但是基于两列的排序对我来说是个问题。请问有人能帮忙吗？

Answer 1

一个非常幼稚的方法，

import tensorflow as tf

a = tf.constant([[1, 2, 3],
                 [2, 3, 5],
                 [1, 4, 6],
                 [2, 2, 1],
                 [0, 4, 2]])

# b = a[:0]*10 + a[:1]*1 -- > (e.g 1*10+2*1 =12)
b = tf.add(tf.slice(a, [0, 0], [-1, 1]) * 10, tf.slice(a, [0, 1], [-1, 1]))

reordered = tf.gather(a, tf.nn.top_k(b[:, 0], k=5, sorted=False).indices)
reordered = tf.reverse(reordered, axis=[0])

with tf.Session() as sess:
    result = sess.run(reordered)
    print(result)

希望这对您有所帮助。