TensorFlow使用哪种排序算法？

Question

我试图弄清楚TensorFlow的排序功能如何工作。

我已经查看了文档，并试图在源代码中找到运气。

函数是tf.contrib.framework.sort（values）吗？ 1

Answer 1

在排序源代码中，this是调用实际排序操作的行。 topk op实际上是核心内核的一部分，并且是用C / C ++编写的。 This是其相应的源代码。快速浏览一下似乎采用了某种形式的堆排序。