如何在张量流中获取矩阵中每一行的最大值和次级最大值？

Question

如何在张量流中获取矩阵中每一行的最大值和次级最大值？例如。我有一个矩阵张量为[[1,2,3]，[6,5,4]，[7,9,8]]。然后，我想获得每行的最大值矢量，例如[3,6,9]和次级值矢量[2,4,8]。问题可能可以分为两部分：1.查找值索引2.从矩阵中收集值并形成一个新向量。

Answer 1

您可能想使用tf.nn.top_k来完成您要寻找的内容：

  

查找最后一个维度的k个最大条目的值和索引。 ...对于矩阵（分别为较高的秩输入），计算每行中的前k个条目（沿最后一维的对应向量）