Sort one list from another list in TensorFlow

Question

I have two tf.Tensors A: [x0, x1, x2, x3, x4] and B: [2, 2, 1, 3, 2]. I would like to sort A using B.

Basically I would like to do the following, but using only TF operators:

list1, list2 = zip(*sorted(zip(list1, list2)))

I tried tf.sort() with tf.stack, but it seem to sort each dimension independently. I think I need to use tf.argsort similarly to this answer Sort array's rows by another array in Python but the indexing fails as tensor indexing do not seems to be supported.

Conchylicultor · Accepted Answer · 2019-07-26 01:10:24Z

1

I think I found the solution:

list1 = [2, 2, 1, 3, 2]
list2 = [0, 1, 2, 3, 4]
ids = tf.argsort(list1)
out = tf.gather(list2, ids)  # [2, 0, 1, 4, 3]

answered Jul 26, 2019 at 1:10

Conchylicultor

5,9893 gold badges47 silver badges48 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Sort one list from another list in TensorFlow

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related