假设我在 TensorFlow 中有一个数组：
[ 0.12300211, 0.51767069, 0.13886075, 0.55363625],
[ 0.47279349, 0.50432992, 0.48080254, 0.51576483],
[ 0.84347934, 0.44505221, 0.88839239, 0.48857492],
[ 0.93650454, 0.43652734, 0.96464157, 0.47236174], ..
我想按第三列对该数组进行排序。我该怎么做呢？我可以使用单独对每一列进行排序tf.nn.top_k()，它给我排序后的值和各自的索引。我可以使用第三列的索引来重新排序其他列，但我找不到重新排序操作。
假设我想将事情保留在图表中（没有 Python 恶作剧）：
- 如何在 TensorFlow 中对（上述数组）进行排序？
- 当我有要重新排序的索引时，如何在 TensorFlow 中重新排序？
以下作品：
a = tf.constant(...) # the array
reordered = tf.gather(a, tf.nn.top_k(a[:, 2], k=4).indices)
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
