在TensorFlow中对数组排序

以下作品：

a = tf.constant(...) # the arrayreordered = tf.gather(a, tf.nn.top_k(a[:, 2], k=4).indices)